Add to train-play test

HorizonRobotics · Dec 19, 2023 · 944496e · 944496e
1 parent 92e40ba
commit 944496e
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/alf/bin/train_play_test.py b/alf/bin/train_play_test.py
@@ -697,6 +697,11 @@ def test_causal_bc_pendulum(self):
             conf_file='./hybrid_rl/causal_bc_pendulum_conf.py',
             extra_train_params=OFF_POLICY_TRAIN_PARAMS)
 
+    def test_smodice_pendulum(self):
+        self._test(
+            conf_file='./smodice_pendulum_conf.py',
+            extra_train_params=OFF_POLICY_TRAIN_PARAMS)
+
     def test_iql_pendulum(self):
         self._test(
             conf_file='./hybrid_rl/iql_pendulum_conf.py',