Minor

Author: MischaPanch

examples/mujoco/mujoco_ppo_hl_multi.py

@@ -28,8 +28,6 @@
 from tianshou.highlevel.params.algorithm_params import PPOParams
 from tianshou.highlevel.params.lr_scheduler import LRSchedulerFactoryFactoryLinear
 
-log = logging.getLogger(__name__)
-
 
 def main(
     num_experiments: int = 5,

tianshou/evaluation/launcher.py

@@ -144,14 +144,14 @@ def _launch(self, experiments: Sequence["Experiment"]) -> list[InfoStats | None]
 
 
 class RegisteredExpLauncher(Enum):
-    joblib = "joblib"
-    sequential = "sequential"
+    JOBLIB = "joblib"
+    SEQUENTIAL = "sequential"
 
     def create_launcher(self) -> ExpLauncher:
         match self:
-            case RegisteredExpLauncher.joblib:
+            case RegisteredExpLauncher.JOBLIB:
                 return JoblibExpLauncher()
-            case RegisteredExpLauncher.sequential:
+            case RegisteredExpLauncher.SEQUENTIAL:
                 return SequentialExpLauncher()
             case _:
                 raise NotImplementedError(

tianshou/highlevel/experiment.py

@@ -490,7 +490,7 @@ def __init__(self, experiments: list[Experiment]):
 
     def run(
         self,
-        launcher: ExpLauncher | RegisteredExpLauncher | str = RegisteredExpLauncher.sequential,
+        launcher: ExpLauncher | RegisteredExpLauncher | str = RegisteredExpLauncher.SEQUENTIAL,
     ) -> list[InfoStats | None]:
         if isinstance(launcher, str):
             launcher = RegisteredExpLauncher[launcher]
@@ -685,7 +685,7 @@ def build_seeded_collection(self, num_experiments: int) -> ExperimentCollection:
     def build_and_run(
         self,
         num_experiments: int = 1,
-        launcher: ExpLauncher | RegisteredExpLauncher | str = RegisteredExpLauncher.sequential,
+        launcher: ExpLauncher | RegisteredExpLauncher | str = RegisteredExpLauncher.SEQUENTIAL,
         perform_rliable_analysis: bool = True,
     ) -> list[InfoStats | None]:
         """Build and run experiments. With multiple experiments, the seeds will be non-overlapping and the parallelism is controlled by the launcher.