fix pybullet visualizer

kscalelabs · Sep 5, 2024 · 7d5850d · 7d5850d
1 parent d2877cb
commit 7d5850d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/kscale/store/pybullet.py b/kscale/store/pybullet.py
@@ -27,7 +27,10 @@ async def main(args: Sequence[str] | None = None) -> None:
     parsed_args = parser.parse_args(args)
 
     # Gets the URDF path.
-    urdf_path = await download_urdf(parsed_args.listing_id)
+    urdf_dir = await download_urdf(parsed_args.listing_id)
+    urdf_path = next(urdf_dir.glob("*.urdf"), None)
+    if urdf_path is None:
+        raise ValueError(f"No URDF found in {urdf_dir}")
 
     try:
         import pybullet as p  # type: ignore[import-not-found]