Envs fixes

P6-Pool · Apr 11, 2024 · ec672ce · ec672ce
1 parent 19eb909
commit ec672ce
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/src/fastfiz_env/envs/__init__.py b/src/fastfiz_env/envs/__init__.py
@@ -4,15 +4,13 @@
 
 from .base_fastfiz import BaseFastFiz
 from .base_rl_fastfiz import BaseRLFastFiz
-from .event_fastfiz import EventFastFiz
 from .velocity_fastfiz import VelocityFastFiz
 from .simple_fastfiz import SimpleFastFiz
 from .testing_fastfiz import TestingFastFiz
 
 __all__ = [
     "BaseFastFiz",
     "BaseRLFastFiz",
-    "EventFastFiz",
     "VelocityFastFiz",
     "SimpleFastFiz",
     "TestingFastFiz",

diff --git a/src/fastfiz_env/envs/simple_fastfiz.py b/src/fastfiz_env/envs/simple_fastfiz.py
@@ -62,7 +62,7 @@ def step(self, action: np.ndarray) -> tuple[np.ndarray, float, bool, bool, dict]
         prev_table_state = ff.TableState(self.table_state)
         shot_params = ff.ShotParams(*action)
 
-        if not self._possible_shot(shot_params):
+        if self._possible_shot(shot_params):
             self.table_state.executeShot(shot_params)
 
         observation = self._get_observation()