Port SpatialSoftmax and remove Robomimic dependency (#182)

Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-05-16 10:34:10 -04:00
parent 68c1b13406
commit c9069df9f1
7 changed files with 75 additions and 179 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -44,7 +44,6 @@ diffusers = "^0.27.2"
 torchvision = ">=0.18.0"
 h5py = ">=3.10.0"
 huggingface-hub = ">=0.21.4"
-robomimic = "0.2.0"
 gymnasium = ">=0.29.1"
 cmake = ">=3.29.0.1"
 gym-pusht = { version = ">=0.1.3", optional = true}