gymnasium, no baselines3

utiasDSL · Apr 16, 2023 · 2026165 · 2026165
1 parent 3ab630e
commit 2026165
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 10 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -17,9 +17,7 @@ scipy = "^1.10"
 matplotlib = "^3.7"
 pytest = "^7.3"
 pybullet = "^3.2.5"
-stable-baselines3 = "^1.8.0"
 gymnasium = "^0.28"
-torch = "^2.0.0"
 
 [tool.poetry.dev-dependencies]
 

diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -1,15 +1,15 @@
-def test_downwash():
-    from gym_pybullet_drones.examples.downwash import run
-    run(gui=False,plot=False,output_folder='tmp')
-
 def test_fly():
     from gym_pybullet_drones.examples.pid import run
     run(gui=False,plot=False,output_folder='tmp')
 
-def test_learn():
-    from gym_pybullet_drones.examples.learning.learn import run
-    run(gui=False,plot=False,output_folder='tmp')
-
 def test_velocity():
     from gym_pybullet_drones.examples.pid_velocity import run
     run(gui=False,plot=False,output_folder='tmp')
+
+def test_downwash():
+    from gym_pybullet_drones.examples.downwash import run
+    run(gui=False,plot=False,output_folder='tmp')
+
+# def test_learn():
+#     from gym_pybullet_drones.examples.learning.learn import run
+#     run(gui=False,plot=False,output_folder='tmp')