@@ -14,7 +14,8 @@ dependencies = [
    "gymnasium[all]>=1.0.0",
    "prpl_utils@git+https://github.com/Princeton-Robot-Planning-and-Learning/prpl-utils.git",
    "prpl_llm_utils@git+https://github.com/Princeton-Robot-Planning-and-Learning/prpl-llm-utils.git",
-   "prbench[all] @ git+https://github.com/Princeton-Robot-Planning-and-Learning/prbench.git@main",   "moviepy",
+   "prbench[geom2d]@git+https://github.com/Princeton-Robot-Planning-and-Learning/prbench.git@c897a36aedaf0721a808ed631fd615b55e1e0b69",
+   "moviepy",
    "pandas",
    "pandas-stubs",
    "hydra-core",