verifiers 0.1.15.dev151__tar.gz → 0.1.15.dev152__tar.gz

{verifiers-0.1.15.dev151 → verifiers-0.1.15.dev152}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: verifiers
-Version: 0.1.15.dev151
+Version: 0.1.15.dev152
 Summary: Verifiers: Environments for LLM Reinforcement Learning
 Project-URL: Homepage, https://github.com/primeintellect-ai/verifiers
 Project-URL: Documentation, https://github.com/primeintellect-ai/verifiers

{verifiers-0.1.15.dev151 → verifiers-0.1.15.dev152}/verifiers/envs/experimental/composable/tasksets/swe/scale_swe/taskset.py

@@ -34,6 +34,7 @@ from verifiers.envs.experimental.composable import SandboxSpec, SandboxTaskSet
 logger = logging.getLogger(__name__)
 
 DATASET_NAME = "AweAI-Team/Scale-SWE"
+REGISTRY_PREFIX = "us-central1-docker.pkg.dev/prime-intellect-platform/prod-sandbox"
 
 ENV_VARS_SCALE_SWE = {
     "PATH": (
@@ -70,6 +71,13 @@ if __name__ == "__main__":
 """
 
 
+def _build_docker_image(info: dict) -> str:
+    image = info["image_url"]
+    if image.startswith(REGISTRY_PREFIX):
+        return image
+    return f"{REGISTRY_PREFIX}/{image}"
+
+
 class ScaleSWERubric(vf.Rubric):
     """Scores Scale-SWE tasks by checking merged F2P/P2P pytest results."""
 
@@ -155,7 +163,7 @@ class ScaleSWETaskSet(SandboxTaskSet):
 
     def get_sandbox_spec(self, info: dict) -> SandboxSpec:
         return SandboxSpec(
-            image=info["image_url"],
+            image=_build_docker_image(info),
             timeout_minutes=self.timeout_minutes,
         )