sublinear 0.2.1__tar.gz → 0.2.2__tar.gz

{sublinear-0.2.1 → sublinear-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sublinear
-Version: 0.2.1
+Version: 0.2.2
 Summary: Python SDK for Sublinear foundation reward models for robotics.
 Author: Sublinear
 License-Expression: MIT

{sublinear-0.2.1 → sublinear-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sublinear"
-version = "0.2.1"
+version = "0.2.2"
 description = "Python SDK for Sublinear foundation reward models for robotics."
 readme = "README.md"
 requires-python = ">=3.10"

{sublinear-0.2.1 → sublinear-0.2.2}/src/sublinear/core.py

@@ -445,11 +445,11 @@ class Sublinear:
         final_video_paths_external_view_iterable = (enumerate(final_video_paths_external_view) if final_video_paths_external_view is not None else repeat(None, video_count))
         final_video_paths_wrist_view_iterable = (enumerate(final_video_paths_wrist_view) if final_video_paths_wrist_view is not None else repeat(None, video_count))
         model = model.lower()
-        if model not in ["sole-r1", "robometer", "roboreward", "topreward"]:
-            raise ValueError(
-                f"Model '{model}' is not supported. "
-                "Supported models are: 'sole-r1', 'robometer', 'roboreward', 'topreward'."
-            )
+        # if "dm" not in model and model not in ["sole-r1", "robometer", "roboreward", "topreward"]:
+        #     raise ValueError(
+        #         f"Model '{model}' is not supported. "
+        #         "Supported models are: 'sole-r1', 'robometer', 'roboreward', 'topreward'."
+        #     )
         active_job_ids: set[str] = set()
         active_job_ids_lock = threading.Lock()
         cancel_requested = threading.Event()

{sublinear-0.2.1 → sublinear-0.2.2}/src/sublinear.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sublinear
-Version: 0.2.1
+Version: 0.2.2
 Summary: Python SDK for Sublinear foundation reward models for robotics.
 Author: Sublinear
 License-Expression: MIT