fix lints

Author: nastaran78

neps/plot/tensorboard_eval.py

@@ -39,9 +39,9 @@ class tblogger:  # noqa: N801
     @classmethod
     def initiate_internal_configurations(
         cls,
-        root_directory: Path | None = None,
-        pipeline_directory: Path | None = None,
-        previous_pipeline_directory: Path | None = None,
+        root_directory: Path | str | None = None,
+        pipeline_directory: Path | str | None = None,
+        previous_pipeline_directory: Path | str | None = None,
     ) -> None:
         """Initialize internal directories and configuration for TensorBoard logging.
 
@@ -71,6 +71,26 @@ def initiate_internal_configurations(
                 if trial.metadata.previous_trial_location
                 else None
             )
+        else:
+            # Convert str paths to Path objects
+            root_directory = (
+                Path(root_directory)
+                if isinstance(root_directory, str)
+                else root_directory
+            )
+            pipeline_directory = (
+                Path(pipeline_directory)
+                if isinstance(pipeline_directory, str)
+                else pipeline_directory
+            )
+            previous_pipeline_directory = (
+                Path(previous_pipeline_directory)
+                if isinstance(previous_pipeline_directory, str)
+                else previous_pipeline_directory
+            )
+
+            if previous_pipeline_directory and not previous_pipeline_directory.exists():
+                previous_pipeline_directory = None
 
         register_notify_trial_end("NEPS_TBLOGGER", cls.end_of_config)
 

neps/runtime.py

@@ -111,10 +111,12 @@ def _set_workers_neps_state(state: NePSState) -> None:
     global _WORKER_NEPS_STATE  # noqa: PLW0603
     _WORKER_NEPS_STATE = state
 
+
 def is_in_progress_trial_set() -> bool:
     """Check if the currently running trial in this process is set."""
     return _CURRENTLY_RUNNING_TRIAL_IN_PROCESS is not None
 
+
 def get_in_progress_trial() -> Trial:
     """Get the currently running trial in this process."""
     if _CURRENTLY_RUNNING_TRIAL_IN_PROCESS is None: