Apply code review suggestions

huggingface · Jun 18, 2024 · c345056 · c345056
1 parent 43b2e26
commit c345056
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 55 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -42,7 +42,6 @@ torch = "^2.2.1"
 opencv-python = ">=4.9.0"
 diffusers = "^0.27.2"
 torchvision = ">=0.17.1"
-# torchvision = { path = "../../build/vision" }
 h5py = ">=3.10.0"
 huggingface-hub = {extras = ["hf-transfer"], version = "^0.23.0"}
 gymnasium = ">=0.29.1"
@@ -72,7 +71,7 @@ aloha = ["gym-aloha"]
 dev = ["pre-commit", "debugpy"]
 test = ["pytest", "pytest-cov"]
 umi = ["imagecodecs"]
-benchmark = ["scikit-image"]
+video_benchmark = ["scikit-image"]
 
 [tool.ruff]
 line-length = 110