Fixing dependencies and added debugging for imports

inference_openpi.py

@@ -17,41 +17,88 @@ os.environ.setdefault("MUJOCO_GL", "egl")
 os.environ.setdefault("PYOPENGL_PLATFORM", "egl")
 os.environ.setdefault("XDG_RUNTIME_DIR", "/tmp")
 
+# Verify critical dependencies are available
+print("===== Checking OpenPI Environment Dependencies =====", file=sys.stderr, flush=True)
+
+try:
+    import roboeval
+    print("✓ roboeval imported successfully", file=sys.stderr, flush=True)
+except ImportError as e:
+    print(f"✗ roboeval import failed: {e}", file=sys.stderr, flush=True)
+    sys.exit(1)
+
+try:
+    import lerobot
+    print("✓ lerobot imported successfully", file=sys.stderr, flush=True)
+except ImportError as e:
+    print(f"✗ lerobot import failed: {e}", file=sys.stderr, flush=True)
+    sys.exit(1)
+
+try:
+    import openpi
+    print("✓ openpi imported successfully", file=sys.stderr, flush=True)
+except ImportError as e:
+    print(f"✗ openpi import failed: {e}", file=sys.stderr, flush=True)
+    sys.exit(1)
+
 # Import OpenPI dependencies (only available in openpi_env)
-from openpi.training.config import Config as PIConfig
-from openpi.policies.policy_config import PIPolicy
-from roboeval.action_modes import JointPositionActionMode
-from roboeval.utils.observation_config import ObservationConfig, CameraConfig
-from roboeval.robots.configs.panda import BimanualPanda
-from roboeval.roboeval_env import CONTROL_FREQUENCY_MAX
+try:
+    from openpi.training.config import Config as PIConfig
+    from openpi.policies.policy_config import PIPolicy
+    print("✓ OpenPI config and policy modules imported successfully", file=sys.stderr, flush=True)
+except ImportError as e:
+    print(f"✗ OpenPI config/policy import failed: {e}", file=sys.stderr, flush=True)
+    sys.exit(1)
+
+try:
+    from roboeval.action_modes import JointPositionActionMode
+    from roboeval.utils.observation_config import ObservationConfig, CameraConfig
+    from roboeval.robots.configs.panda import BimanualPanda
+    from roboeval.roboeval_env import CONTROL_FREQUENCY_MAX
+    print("✓ RoboEval core modules imported successfully", file=sys.stderr, flush=True)
+except ImportError as e:
+    print(f"✗ RoboEval core modules import failed: {e}", file=sys.stderr, flush=True)
+    sys.exit(1)
 
 # Import all environment classes
-from roboeval.envs.manipulation import (
-    CubeHandover, CubeHandoverOrientation, CubeHandoverPosition,
-    CubeHandoverPositionAndOrientation, VerticalCubeHandover,
-    StackTwoBlocks, StackTwoBlocksOrientation, StackTwoBlocksPosition,
-    StackTwoBlocksPositionAndOrientation
-)
-from roboeval.envs.lift_pot import (
-    LiftPot, LiftPotOrientation, LiftPotPosition, LiftPotPositionAndOrientation,
-)
-from roboeval.envs.lift_tray import (
-    LiftTray, DragOverAndLiftTray, LiftTrayOrientation, LiftTrayPosition, LiftTrayPositionAndOrientation,
-)
-from roboeval.envs.pack_objects import (
-    PackBox, PackBoxOrientation, PackBoxPosition, PackBoxPositionAndOrientation,
-)
-from roboeval.envs.stack_books import (
-    PickSingleBookFromTable, PickSingleBookFromTableOrientation,
-    PickSingleBookFromTablePosition, PickSingleBookFromTablePositionAndOrientation,
-    StackSingleBookShelf, StackSingleBookShelfPosition, StackSingleBookShelfPositionAndOrientation,
-)
-from roboeval.envs.rotate_utility_objects import (
-    RotateValve, RotateValveObstacle, RotateValvePosition, RotateValvePositionAndOrientation,
-)
+try:
+    from roboeval.envs.manipulation import (
+        CubeHandover, CubeHandoverOrientation, CubeHandoverPosition,
+        CubeHandoverPositionAndOrientation, VerticalCubeHandover,
+        StackTwoBlocks, StackTwoBlocksOrientation, StackTwoBlocksPosition,
+        StackTwoBlocksPositionAndOrientation
+    )
+    from roboeval.envs.lift_pot import (
+        LiftPot, LiftPotOrientation, LiftPotPosition, LiftPotPositionAndOrientation,
+    )
+    from roboeval.envs.lift_tray import (
+        LiftTray, DragOverAndLiftTray, LiftTrayOrientation, LiftTrayPosition, LiftTrayPositionAndOrientation,
+    )
+    from roboeval.envs.pack_objects import (
+        PackBox, PackBoxOrientation, PackBoxPosition, PackBoxPositionAndOrientation,
+    )
+    from roboeval.envs.stack_books import (
+        PickSingleBookFromTable, PickSingleBookFromTableOrientation,
+        PickSingleBookFromTablePosition, PickSingleBookFromTablePositionAndOrientation,
+        StackSingleBookShelf, StackSingleBookShelfPosition, StackSingleBookShelfPositionAndOrientation,
+    )
+    from roboeval.envs.rotate_utility_objects import (
+        RotateValve, RotateValveObstacle, RotateValvePosition, RotateValvePositionAndOrientation,
+    )
+    print("✓ RoboEval environment classes imported successfully", file=sys.stderr, flush=True)
+except ImportError as e:
+    print(f"✗ RoboEval environment classes import failed: {e}", file=sys.stderr, flush=True)
+    sys.exit(1)
 
 # Video
-from moviepy.editor import VideoClip
+try:
+    from moviepy.editor import VideoClip
+    print("✓ moviepy imported successfully", file=sys.stderr, flush=True)
+except ImportError as e:
+    print(f"✗ moviepy import failed: {e}", file=sys.stderr, flush=True)
+    sys.exit(1)
+
+print("===== All dependencies verified successfully =====", file=sys.stderr, flush=True)
 
 # Environment registry
 _ENV_CLASSES = {

requirements.txt

@@ -5,7 +5,9 @@
 gradio>=4.0.0
 numpy>=1.22.4,<2.0.0
 pillow>=11.0.0
-huggingface-hub>=0.36.0
+# Pin huggingface-hub to version compatible with Gradio (HfFolder was removed in 0.26.0+)
+# Gradio 4.x requires huggingface-hub < 0.26.0
+huggingface-hub>=0.20.0,<0.26.0
 
 # Note: OpenPI and OpenVLA dependencies are installed in separate conda environments
 # See requirements_openpi.txt and requirements_openvla.txt

requirements_openpi.txt

@@ -1,8 +1,6 @@
 # OpenPI environment dependencies
-# PyTorch 2.7.0 for OpenPI
-torch==2.7.0
-torchvision==0.20.0
-torchaudio==2.7.0
+# PyTorch 2.7.0+ for OpenPI (working version from user's tested setup)
+torch>=2.7.0
 
 # JAX and ML utilities
 jax[cuda12]==0.5.3
@@ -39,11 +37,10 @@ transformers==4.48.1
 rich>=14.0.0
 polars>=1.30.0
 
-# Hugging Face
+# Hugging Face (simplified versions)
 datasets==3.6.0
-huggingface-hub>=0.36.0
+huggingface-hub==0.34.4
 diffusers==0.26.3
-tokenizers>=0.19.1
 
 # Utilities
 hydra-core==1.3.2

setup.sh

@@ -79,5 +79,5 @@ echo "OpenPI environment ready"
 echo ""
 echo "===== Setup Complete ====="
 echo "✓ Base environment: Gradio + RoboEval"
-echo "✓ openpi_env: PyTorch 2.7 + OpenPI"
+echo "✓ openpi_env: PyTorch 2.7+ + OpenPI"
 echo "ℹ openvla_env: Disabled (uncomment in setup.sh to enable)"