small number of changes to install

Emerge-Lab · May 7, 2024 · 2a74f35 · 2a74f35
1 parent 870252a
commit 2a74f35
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/algorithms/ppo/sb3/mappo.py b/algorithms/ppo/sb3/mappo.py
@@ -1,9 +1,7 @@
 import logging
-from typing import Optional
 import torch
 from torch.nn import functional as F
 import numpy as np
-from box import Box
 from gymnasium import spaces
 from stable_baselines3 import PPO
 from stable_baselines3.common.callbacks import BaseCallback

diff --git a/environment.yml b/environment.yml
@@ -54,12 +54,14 @@ dependencies:
       - poetry-plugin-export==1.7.1
       - ptyprocess==0.7.0
       - pycparser==2.22
+      - pygame
       - pyproject-hooks==1.0.0
       - rapidfuzz==3.7.0
       - requests==2.31.0
       - requests-toolbelt==1.0.0
       - secretstorage==3.3.3
       - shellingham==1.5.4
+      - stable-baselines3==2.3.2
       - tomlkit==0.12.4
       - trove-classifiers==2024.3.25
       - urllib3==2.2.1

diff --git a/pyproject.toml b/pyproject.toml
@@ -30,5 +30,6 @@ description = ""
 authors = ["Sébastien Eustace <[email protected]>"]
 readme = "README.md"
 packages = [
-    {include = "gpudrive.cpython-311-x86_64-linux-gnu.so", from = "build", to = "gpudrive"}
-]
+    {include = "gpudrive.cpython-311-x86_64-linux-gnu.so", from = "build", to = "gpudrive"},
+    {include = "pygpudrive"}
+]