diff --git a/alf/bin/train_play_test.py b/alf/bin/train_play_test.py index a0d2e4208..a02a9c86f 100644 --- 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',