kscale 0.3.6__py3-none-any.whl → 0.3.7__py3-none-any.whl

kscale/__init__.py

@@ -1,6 +1,6 @@
 """Defines the common interface for the K-Scale Python API."""
 
-__version__ = "0.3.6"
+__version__ = "0.3.7"
 
 from pathlib import Path
 

kscale/requirements.txt

@@ -21,6 +21,3 @@ tabulate
 
 # Async
 async-lru
-
-# K-Scale
-krec

kscale/web/cli/robot_class.py

@@ -80,7 +80,23 @@ async def update(current_name: str, name: str | None = None, description: str |
     click.echo(f"  Description: {click.style(robot_class.description or 'N/A', fg='yellow')}")
 
 
-@cli.command()
+@cli.command("delete")
+@click.argument("name")
+@coro
+async def delete_robot_class(name: str) -> None:
+    """Deletes a robot class."""
+    async with RobotClassClient() as client:
+        await client.delete_robot_class(name)
+    click.echo(f"Robot class deleted: {click.style(name, fg='red')}")
+
+
+@cli.group()
+def metadata() -> None:
+    """Handle the robot class metadata."""
+    pass
+
+
+@metadata.command("update")
 @click.argument("name")
 @click.argument("json_path", type=click.Path(exists=True))
 @coro
@@ -96,7 +112,7 @@ async def update_metadata(name: str, json_path: str) -> None:
     click.echo(f"  Name: {click.style(robot_class.class_name, fg='green')}")
 
 
-@cli.command()
+@metadata.command("get")
 @click.argument("name")
 @click.option("--json-path", type=click.Path(exists=False))
 @coro
@@ -115,27 +131,17 @@ async def get_metadata(name: str, json_path: str | None = None) -> None:
             json.dump(metadata.model_dump(), f)
 
 
-@cli.command()
-@click.argument("name")
-@coro
-async def delete(name: str) -> None:
-    """Deletes a robot class."""
-    async with RobotClassClient() as client:
-        await client.delete_robot_class(name)
-    click.echo(f"Robot class deleted: {click.style(name, fg='red')}")
-
-
 @cli.group()
 def urdf() -> None:
     """Handle the robot class URDF."""
     pass
 
 
-@urdf.command()
+@urdf.command("upload")
 @click.argument("class_name")
 @click.argument("urdf_file")
 @coro
-async def upload(class_name: str, urdf_file: str) -> None:
+async def upload_urdf(class_name: str, urdf_file: str) -> None:
     """Uploads a URDF file to a robot class."""
     async with RobotClassClient() as client:
         response = await client.upload_robot_class_urdf(class_name, urdf_file)
@@ -143,18 +149,18 @@ async def upload(class_name: str, urdf_file: str) -> None:
     click.echo(f"  Filename: {click.style(response.filename, fg='green')}")
 
 
-@urdf.command()
+@urdf.command("download")
 @click.argument("class_name")
 @click.option("--cache", is_flag=True, default=False)
 @coro
-async def download(class_name: str, cache: bool) -> None:
+async def download_urdf(class_name: str, cache: bool) -> None:
     """Downloads a URDF file from a robot class."""
     async with RobotClassClient() as client:
         urdf_file = await client.download_and_extract_urdf(class_name, cache=cache)
     click.echo(f"URDF downloaded: {click.style(urdf_file, fg='green')}")
 
 
-@urdf.command()
+@urdf.command("pybullet")
 @click.argument("class_name")
 @click.option("--no-cache", is_flag=True, default=False)
 @click.option("--hide-gui", is_flag=True, default=False)
@@ -168,7 +174,7 @@ async def download(class_name: str, cache: bool) -> None:
 @click.option("--start-height", type=float, default=0.0)
 @click.option("--cycle-duration", type=float, default=2.0)
 @coro
-async def pybullet(
+async def run_pybullet(
     class_name: str,
     no_cache: bool,
     hide_gui: bool,
@@ -432,11 +438,11 @@ async def pybullet(
         last_time = cur_time
 
 
-@urdf.command()
+@urdf.command("mujoco")
 @click.argument("class_name")
 @click.option("--no-cache", is_flag=True, default=False)
 @coro
-async def mujoco(class_name: str, no_cache: bool) -> None:
+async def run_mujoco(class_name: str, no_cache: bool) -> None:
     """Shows the URDF file for a robot class in Mujoco.
 
     This command downloads and extracts the robot class URDF folder,
@@ -444,11 +450,11 @@ async def mujoco(class_name: str, no_cache: bool) -> None:
     launches the Mujoco viewer using the provided MJCF file.
     """
     try:
-        from mujoco.viewer import launch_from_path
+        import mujoco.viewer
     except ImportError:
         click.echo(
             click.style(
-                "Mujoco and mujoco-python-viewer are required; install with `pip install mujoco mujoco-python-viewer`",
+                "Mujoco and mujoco-python-viewer are required; install with `pip install mujoco`",
                 fg="red",
             )
         )
@@ -463,8 +469,9 @@ async def mujoco(class_name: str, no_cache: bool) -> None:
         click.echo(click.style(f"No MJCF file found in {extracted_folder}", fg="red"))
         return
 
-    click.echo(f"Launching Mujoco viewer with: {click.style(str(mjcf_file.resolve()), fg='green')}")
-    launch_from_path(str(mjcf_file.resolve()))
+    mjcf_path_str = str(mjcf_file.resolve())
+    click.echo(f"Launching Mujoco viewer with: {click.style(mjcf_path_str, fg='green')}")
+    mujoco.viewer.launch_from_path(mjcf_path_str)
 
 
 if __name__ == "__main__":

{kscale-0.3.6.dist-info → kscale-0.3.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: kscale
-Version: 0.3.6
+Version: 0.3.7
 Summary: The kscale project
 Home-page: https://github.com/kscalelabs/kscale
 Author: Benjamin Bolte
@@ -21,7 +21,6 @@ Requires-Dist: click
 Requires-Dist: colorlogging
 Requires-Dist: tabulate
 Requires-Dist: async-lru
-Requires-Dist: krec
 Provides-Extra: dev
 Requires-Dist: black; extra == "dev"
 Requires-Dist: darglint; extra == "dev"

{kscale-0.3.6.dist-info → kscale-0.3.7.dist-info}/RECORD

@@ -1,9 +1,9 @@
-kscale/__init__.py,sha256=ny2947fOKliAIcJrudjMn6Ocfzk9Hb-VXUNFPikAPGo,200
+kscale/__init__.py,sha256=JXk1-68D39Xj5C7awsV8wBXSDoj4Zds3jDuQP0VvMyY,200
 kscale/cli.py,sha256=PMHLKR5UwdbbReVmqHXpJ-K9-mGHv_0I7KQkwxmFcUA,881
 kscale/conf.py,sha256=dm35XSnzJp93St-ixVtYN4Nvqvb5upPGBrWkSI6Yb-4,1743
 kscale/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kscale/requirements-dev.txt,sha256=WI7-ea4IRJakmqVMN8QKhOsDGrghwtvk03aIsFaNSIw,130
-kscale/requirements.txt,sha256=_BGbnKTQaXKx0bNEG0wguod9swsiCb2mF6rLm7sFJ2Q,214
+kscale/requirements.txt,sha256=tj8wBkIhl6ijk5rFqtPCdkpdOS7XSIcUvsz_Eq1axcw,198
 kscale/artifacts/__init__.py,sha256=RK8wdybtCJPgdLLJ8R8-YMi1Ph5ojqAKVJZowHONtgo,232
 kscale/artifacts/plane.obj,sha256=x59-IIrWpLjhotChiqT2Ul6U8s0RcHkaEeUZb4KXL1c,348
 kscale/artifacts/plane.urdf,sha256=LCiTk14AyTHjkZ1jvsb0hNaEaJUxDb8Z1JjsgpXu3YM,819
@@ -15,7 +15,7 @@ kscale/web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kscale/web/utils.py,sha256=Mme-FAQ0_zbjjOQeX8wyq8F4kL4i9fH7ytri16U6qOA,1046
 kscale/web/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kscale/web/cli/robot.py,sha256=rI-A4_0uvJPeA71Apl4Z3mV5fIfWkgmzT9JRmJYxz3A,3307
-kscale/web/cli/robot_class.py,sha256=Ymyrc3UqsO7PzP_wJFXEWW37knMCDZTYP-8ziXVa-AQ,18710
+kscale/web/cli/robot_class.py,sha256=QjE65uw2lSaerrdFr_zZzN9_W5w4Rx4exWZkijpDzmQ,18892
 kscale/web/cli/token.py,sha256=1rFC8MYKtqbNsQa2KIqwW1tqpaMtFaxuNsallwejXTU,787
 kscale/web/cli/user.py,sha256=aaJJCL1P5lfhK6ZC9OwOHXKA-I3MWqVZ_k7TYnx33CY,1303
 kscale/web/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -26,9 +26,9 @@ kscale/web/clients/robot_class.py,sha256=G8Nk6V7LGJE9Wpg9tyyCkIfz1fRTsxXQRgHtlei
 kscale/web/clients/user.py,sha256=jsa1_s6qXRM-AGBbHlPhd1NierUtynjY9tVAPNr6_Os,568
 kscale/web/gen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kscale/web/gen/api.py,sha256=VWlt8TaMZaAhCPPIWhx1d4yE0BRfpVz0-_Cps8hjgAI,4662
-kscale-0.3.6.dist-info/LICENSE,sha256=HCN2bImAzUOXldAZZI7JZ9PYq6OwMlDAP_PpX1HnuN0,1071
-kscale-0.3.6.dist-info/METADATA,sha256=IHjwsym9rqmZnPltfYE3nGywHbDFR-HCOVfh_YrG_Bg,2340
-kscale-0.3.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-kscale-0.3.6.dist-info/entry_points.txt,sha256=N_0pCpPnwGDYVzOeuaSOrbJkS5L3lS9d8CxpJF1f8UI,62
-kscale-0.3.6.dist-info/top_level.txt,sha256=C2ynjYwopg6YjgttnI2dJjasyq3EKNmYp-IfQg9Xms4,7
-kscale-0.3.6.dist-info/RECORD,,
+kscale-0.3.7.dist-info/LICENSE,sha256=HCN2bImAzUOXldAZZI7JZ9PYq6OwMlDAP_PpX1HnuN0,1071
+kscale-0.3.7.dist-info/METADATA,sha256=0hDh7EJaKk94ofN-EfGPVcvlSNs1vfcuVTI5vx9DB6Y,2320
+kscale-0.3.7.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+kscale-0.3.7.dist-info/entry_points.txt,sha256=N_0pCpPnwGDYVzOeuaSOrbJkS5L3lS9d8CxpJF1f8UI,62
+kscale-0.3.7.dist-info/top_level.txt,sha256=C2ynjYwopg6YjgttnI2dJjasyq3EKNmYp-IfQg9Xms4,7
+kscale-0.3.7.dist-info/RECORD,,