tracebloc 1.2.77__tar.gz → 1.2.78__tar.gz

{tracebloc-1.2.77 → tracebloc-1.2.78}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tracebloc
-Version: 1.2.77
+Version: 1.2.78
 Summary: Build AI with anyone. On data that can't move. SDK for the tracebloc collaborative AI workspace.
 Author-email: tracebloc <lukas@tracebloc.io>
 License-Expression: MIT

{tracebloc-1.2.77 → tracebloc-1.2.78}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tracebloc"
-version = "1.2.77"
+version = "1.2.78"
 description = "Build AI with anyone. On data that can't move. SDK for the tracebloc collaborative AI workspace."
 license = "MIT"
 license-files = ["license.txt"]

{tracebloc-1.2.77 → tracebloc-1.2.78}/tracebloc/convert/outputs.py

@@ -33,6 +33,8 @@ from tracebloc.validation.header_rules import SHAPE_VALUES
 #: The probe's batch, and its sequence length's ceiling.
 PROBE_BATCH = 2
 PROBE_TOKENS = 16
+#: The random state every forward on the probe starts from.
+PROBE_SEED = 4321
 
 IMAGE_TASKS = frozenset(SHAPE_VALUES["image_size"])
 FEATURE_TASKS = frozenset(SHAPE_VALUES["num_feature_points"]) - IMAGE_TASKS
@@ -137,7 +139,18 @@ class Probe:
     shown: str
 
     def call(self, model: Any) -> Any:
-        """``model`` on the probe, keyword arguments filtered to what it takes."""
+        """``model`` on the probe, keyword arguments filtered to what it takes.
+
+        Each call starts from :data:`PROBE_SEED`, so a forward that draws random
+        numbers in eval (random features, sampling) draws the same on both sides.
+        """
+        import torch
+
+        with torch.random.fork_rng(devices=[]):
+            torch.manual_seed(PROBE_SEED)
+            return self._call(model)
+
+    def _call(self, model: Any) -> Any:
         if not self.named:
             return model(self.tensor)
         try:

{tracebloc-1.2.77 → tracebloc-1.2.78}/tracebloc/convert/training_versions.py

@@ -174,9 +174,9 @@ def refuse_code_failure(
     installed here at another version than training's, that is the refusal.
     """
     libraries = imported_libraries(sources) or ("torch",)
-    clauses = version_mismatches(libraries)
-    if clauses:
-        return ConversionRefusedError(RefusalCause.VERSIONS, refusal_message(clauses))
+    mismatched = version_mismatches(libraries)
+    if mismatched:
+        return ConversionRefusedError(RefusalCause.VERSIONS, refusal_message(mismatched))
     pins = read_pins()
     named = ", ".join(f"{name} {pinned_version(name, pins)}" for name in libraries)
     detail = " ".join(_SOURCE_FILE.sub("", format_exception(error)).split())[:300]

{tracebloc-1.2.77 → tracebloc-1.2.78}/tracebloc.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tracebloc
-Version: 1.2.77
+Version: 1.2.78
 Summary: Build AI with anyone. On data that can't move. SDK for the tracebloc collaborative AI workspace.
 Author-email: tracebloc <lukas@tracebloc.io>
 License-Expression: MIT