kscale 0.0.6__tar.gz → 0.0.7__tar.gz

{kscale-0.0.6/kscale.egg-info → kscale-0.0.7}/PKG-INFO

@@ -1,16 +1,17 @@
 Metadata-Version: 2.1
 Name: kscale
-Version: 0.0.6
+Version: 0.0.7
 Summary: The kscale project
 Home-page: https://github.com/kscalelabs/kscale
 Author: Benjamin Bolte
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: omegaconf==2.3.0
-Requires-Dist: email_validator==2.2.0
-Requires-Dist: httpx==0.27.0
-Requires-Dist: requests==2.32.2
+Requires-Dist: omegaconf
+Requires-Dist: email_validator
+Requires-Dist: httpx
+Requires-Dist: requests
+Requires-Dist: pydantic
 Provides-Extra: dev
 Requires-Dist: black; extra == "dev"
 Requires-Dist: darglint; extra == "dev"

{kscale-0.0.6 → kscale-0.0.7}/kscale/__init__.py

@@ -1,5 +1,5 @@
 """Defines the common interface for the K-Scale Python API."""
 
-__version__ = "0.0.6"
+__version__ = "0.0.7"
 
 from kscale.api import KScale

kscale-0.0.7/kscale/requirements.txt

@@ -0,0 +1,10 @@
+# requirements.txt
+
+# Configuration
+omegaconf
+email_validator
+
+# HTTP requests
+httpx
+requests
+pydantic

{kscale-0.0.6 → kscale-0.0.7}/kscale/store/client.py

@@ -26,6 +26,7 @@ class KScaleStoreClient:
         self.client = httpx.AsyncClient(
             base_url=self.base_url,
             headers={"Authorization": f"Bearer {get_api_key()}"},
+            timeout=httpx.Timeout(30.0),
         )
 
     async def _request(

{kscale-0.0.6 → kscale-0.0.7}/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]

{kscale-0.0.6 → kscale-0.0.7}/kscale/store/urdf.py

@@ -21,6 +21,17 @@ from kscale.store.utils import get_api_key
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+ALLOWED_SUFFIXES = {
+    ".urdf",
+    ".mjcf",
+    ".stl",
+    ".obj",
+    ".dae",
+    ".png",
+    ".jpg",
+    ".jpeg",
+}
+
 
 def get_cache_dir() -> Path:
     return Path(Settings.load().store.cache_dir).expanduser().resolve()
@@ -75,7 +86,7 @@ def create_tarball(folder_path: Path, output_filename: str, cache_dir: Path) ->
     tarball_path = cache_dir / output_filename
     with tarfile.open(tarball_path, "w:gz") as tar:
         for file_path in folder_path.rglob("*"):
-            if file_path.is_file() and file_path.suffix.lower() in (".urdf", ".mjcf", ".stl", ".obj", ".dae"):
+            if file_path.is_file() and file_path.suffix.lower() in ALLOWED_SUFFIXES:
                 tar.add(file_path, arcname=file_path.relative_to(folder_path))
                 logger.info("Added %s to tarball", file_path)
             else:

{kscale-0.0.6 → kscale-0.0.7/kscale.egg-info}/PKG-INFO

@@ -1,16 +1,17 @@
 Metadata-Version: 2.1
 Name: kscale
-Version: 0.0.6
+Version: 0.0.7
 Summary: The kscale project
 Home-page: https://github.com/kscalelabs/kscale
 Author: Benjamin Bolte
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: omegaconf==2.3.0
-Requires-Dist: email_validator==2.2.0
-Requires-Dist: httpx==0.27.0
-Requires-Dist: requests==2.32.2
+Requires-Dist: omegaconf
+Requires-Dist: email_validator
+Requires-Dist: httpx
+Requires-Dist: requests
+Requires-Dist: pydantic
 Provides-Extra: dev
 Requires-Dist: black; extra == "dev"
 Requires-Dist: darglint; extra == "dev"

kscale-0.0.7/kscale.egg-info/requires.txt

@@ -0,0 +1,13 @@
+omegaconf
+email_validator
+httpx
+requests
+pydantic
+
+[dev]
+black
+darglint
+mypy
+pytest
+ruff
+datamodel-code-generator

kscale-0.0.6/kscale/requirements.txt

@@ -1,9 +0,0 @@
-# requirements.txt
-
-# Configuration
-omegaconf==2.3.0
-email_validator==2.2.0
-
-# HTTP requests
-httpx==0.27.0
-requests==2.32.2

kscale-0.0.6/kscale.egg-info/requires.txt

@@ -1,12 +0,0 @@
-omegaconf==2.3.0
-email_validator==2.2.0
-httpx==0.27.0
-requests==2.32.2
-
-[dev]
-black
-darglint
-mypy
-pytest
-ruff
-datamodel-code-generator