update dependencies

dfki-ric-underactuated-lab · Jan 10, 2024 · 788884b · 788884b
1 parent ab5e324
commit 788884b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/.github/workflows/test_ub22.yml b/.github/workflows/test_ub22.yml
@@ -59,13 +59,13 @@ jobs:
        if: env.run_tests == 'true'
        run: |
          sudo apt-get install curl
-     - name: Install drake dependencies
+     - name: Install dependencies
        if: env.run_tests == 'true'
        run: |
          sudo apt-get update
          sudo apt-get install --no-install-recommends \
-           libpython3.10 libx11-6 libsm6 libxt6 libglib2.0-0
-     - name: Install project dependencies
+           libpython3.10 libx11-6 libsm6 libxt6 libglib2.0-0 python3-tk
+     - name: Install project
        if: env.run_tests == 'true'
        run: |
          make install

diff --git a/software/python/setup.py b/software/python/setup.py
@@ -24,6 +24,8 @@
             "drake",
             "crocoddyl",
             "lxml",
+            "gym",
+            "shimmy",
             "tensorflow>=2.6.0",
             "stable_baselines3",
             "sphinx",
@@ -34,8 +36,8 @@
         ],
         "drake": ["drake"],
         "crocoddyl": ["crocoddyl", "lxml"],
-        "tensorflow": ["tensorflow>=2.6.0"],
-        "stable_baselines": ["stable_baselines3"],
+        "tensorflow": ["tensorflow>=2.6.0", "gym", "shimmy"],
+        "stable_baselines": ["stable_baselines3", "gym", "shimmy"],
         "doc": ["sphinx", "sphinx-rtd-theme", "numpydoc"],
         "test": ["pytest", "lark"],
     },