Move rl2env under rl2 module

ahtsan · ahtsan · commit 04f06c6e4322 · 2020-03-03T11:53:22.000-08:00
diff --git a/src/garage/envs/__init__.py b/src/garage/envs/__init__.py
@@ -8,10 +8,9 @@
 from garage.envs.half_cheetah_vel_env import HalfCheetahVelEnv
 from garage.envs.normalized_env import normalize
 from garage.envs.point_env import PointEnv
-from garage.envs.rl2_env import RL2Env
 from garage.envs.task_onehot_wrapper import TaskOnehotWrapper
 
 __all__ = [
     'GarageEnv', 'Step', 'EnvSpec', 'GridWorldEnv', 'HalfCheetahDirEnv',
-    'HalfCheetahVelEnv', 'normalize', 'PointEnv', 'RL2Env', 'TaskOnehotWrapper'
+    'HalfCheetahVelEnv', 'normalize', 'PointEnv', 'TaskOnehotWrapper'
 ]
diff --git a/src/garage/tf/algos/rl2/__init__.py b/src/garage/tf/algos/rl2/__init__.py
@@ -0,0 +1,4 @@
+"""Module for RL2."""
+from garage.tf.algos.rl2.rl2env import RL2Env
+
+__all__ = ['RL2Env']
diff --git a/src/garage/tf/algos/rl2/rl2env.py b/src/garage/tf/algos/rl2/rl2env.py
diff --git a/tests/garage/envs/test_rl2_env.py b/tests/garage/envs/test_rl2_env.py
@@ -1,5 +1,5 @@
 from garage.envs import PointEnv
-from garage.envs import RL2Env
+from garage.tf.algos.rl2 import RL2Env
 
 
 class TestRL2Env:
