fix: prefix all TrainCheck log output with [TrainCheck]

Update logging format strings in all three CLI entry points so every
log message is visually identifiable as coming from TrainCheck, not
the user's training script.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: Essoz

traincheck/checker.py

@@ -263,6 +263,9 @@ def main():
     # INFO logs stay in the log file only.
     stream_handler = logging.StreamHandler()
     stream_handler.setLevel(logging.WARNING)
+    stream_handler.setFormatter(
+        logging.Formatter("[TrainCheck] %(levelname)s: %(message)s")
+    )
     logger.addHandler(stream_handler)
 
     results_by_trace: list[tuple[str, list[CheckerResult]]] = []

traincheck/collect_trace.py

@@ -432,11 +432,12 @@ def main():
             args.warm_up_steps = config.INSTRUMENTATION_POLICY["warm_up"]
 
     # set up logging
+    _log_fmt = "[TrainCheck] %(levelname)s: %(message)s"
     if args.debug_mode:
-        logging.basicConfig(level=logging.DEBUG)
+        logging.basicConfig(level=logging.DEBUG, format=_log_fmt)
         os.environ["TRAINCHECK_DEBUG"] = "1"
     else:
-        logging.basicConfig(level=logging.INFO)
+        logging.basicConfig(level=logging.INFO, format=_log_fmt)
 
     logger = logging.getLogger(__name__)
 

traincheck/infer_engine.py

@@ -285,7 +285,7 @@ def main():
     logging.basicConfig(
         filename=f'traincheck_infer_engine_{datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")}_{pid}.log',
         level=log_level,
-        format="%(asctime)s - %(levelname)s - [%(filename)s:%(lineno)s - %(funcName)20s()] - %(message)s",
+        format="%(asctime)s - [TrainCheck] %(levelname)s - [%(filename)s:%(lineno)s - %(funcName)20s()] - %(message)s",
     )
 
     disabled_relations: list[Relation] = []