Added missing requirements for semantic_sensor

leggedrobotics · Aug 20, 2023 · 71ddbd3 · 71ddbd3
1 parent 6f3a002
commit 71ddbd3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/sensor_processing/semantic_sensor/setup.py b/sensor_processing/semantic_sensor/setup.py
@@ -5,6 +5,10 @@
     packages=[
         "semantic_sensor",
     ],
+    install_requires=[
+        "torch",
+        "torchvision",
+    ],
     package_dir={"": "script"},
 )