Cookings and cleanup

P6-Pool · Apr 10, 2024 · 584be1b · 584be1b
1 parent 3940c6c
commit 584be1b
Show file tree

Hide file tree

Showing 14 changed files with 103 additions and 755 deletions.
diff --git a/src/fastfiz_env/__init__.py b/src/fastfiz_env/__init__.py
@@ -44,16 +44,6 @@
     entry_point="fastfiz_env.envs:BaseRLFastFiz",
 )
 
-register(
-    id="PocketRLFastFiz-v0",
-    entry_point="fastfiz_env.envs:PocketRLFastFiz",
-)
-
-
-register(
-    id="EventFastFiz-v0",
-    entry_point="fastfiz_env.envs:EventFastFiz",
-)
 
 register(
     id="VelocityFastFiz-v0",
@@ -63,25 +53,15 @@
 register(
     id="SimpleFastFiz-v0",
     entry_point="fastfiz_env.envs:SimpleFastFiz",
-)
-
-register(
-    id="BasicRLFastFiz-v0",
-    entry_point="fastfiz_env.envs:BasicRLFastFiz",
-)
-
-
-register(
-    id="TestingFastFiz-v0",
-    entry_point="fastfiz_env.envs:TestingFastFiz",
     additional_wrappers=(
         utils.wrappers.MaxEpisodeStepsInjectionWrapper.wrapper_spec(),
     ),
 )
 
+
 register(
-    id="ActionFastFiz-v0",
-    entry_point="fastfiz_env.envs:ActionFastFiz",
+    id="TestingFastFiz-v0",
+    entry_point="fastfiz_env.envs:TestingFastFiz",
     additional_wrappers=(
         utils.wrappers.MaxEpisodeStepsInjectionWrapper.wrapper_spec(),
     ),

diff --git a/src/fastfiz_env/envs/__init__.py b/src/fastfiz_env/envs/__init__.py
@@ -4,22 +4,16 @@
 
 from .base_fastfiz import BaseFastFiz
 from .base_rl_fastfiz import BaseRLFastFiz
-from .pocket_rl_fastfiz import PocketRLFastFiz
 from .event_fastfiz import EventFastFiz
 from .velocity_fastfiz import VelocityFastFiz
 from .simple_fastfiz import SimpleFastFiz
-from .basic_rl_fastfiz import BasicRLFastFiz
 from .testing_fastfiz import TestingFastFiz
-from .action_fastfiz import ActionFastFiz
 
 __all__ = [
     "BaseFastFiz",
     "BaseRLFastFiz",
-    "PocketRLFastFiz",
     "EventFastFiz",
     "VelocityFastFiz",
     "SimpleFastFiz",
-    "BasicRLFastFiz",
     "TestingFastFiz",
-    "ActionFastFiz",
 ]
diff --git a/src/fastfiz_env/envs/action_fastfiz.py b/src/fastfiz_env/envs/action_fastfiz.py