pyright type check fixes

Author: dtronmans

luxonis_train/core/core.py

@@ -152,11 +152,13 @@ def __init__(
 
         rich.traceback.install(suppress=[pl, torch], show_locals=False)
 
+        tracker_params = self.cfg.tracker.model_dump()
+        tracker_params.pop("_auto_finalize", None)
         self.tracker = LuxonisTrackerPL(
             rank=rank_zero_only.rank,
             mlflow_tracking_uri=self.environ.MLFLOW_TRACKING_URI,
+            **tracker_params,
             _auto_finalize=False,
-            **self.cfg.tracker.model_dump(),
         )
 
         self.run_save_dir = (

luxonis_train/core/utils/infer_utils.py

@@ -148,7 +148,8 @@ def infer_from_loader(
     """
     if save_dir is not None:
         writer = InferenceSaveWriter(Path(save_dir), img_paths)
-        model.pl_trainer.callbacks.append(writer)
+        callbacks = cast(list[Any], model.pl_trainer.callbacks)
+        callbacks.append(writer)
         try:
             model.pl_trainer.predict(
                 model.lightning_module,
@@ -157,7 +158,7 @@ def infer_from_loader(
             )
         finally:
             with suppress(ValueError):
-                model.pl_trainer.callbacks.remove(writer)
+                callbacks.remove(writer)
         return
 
     predictions = model.pl_trainer.predict(model.lightning_module, loader)