code-loader 1.0.205.dev0__tar.gz → 1.0.205.dev1__tar.gz

{code_loader-1.0.205.dev0 → code_loader-1.0.205.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.205.dev0
+Version: 1.0.205.dev1
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT

{code_loader-1.0.205.dev0 → code_loader-1.0.205.dev1}/code_loader/contract/responsedataclasses.py

@@ -86,6 +86,8 @@ class DatasetSetup:
     custom_losses: List[CustomLossInstance]
     metrics: List[MetricInstance] = field(default_factory=list)
     simulations: List[SimulationInstance] = field(default_factory=list)
+    # Structural: the integration registered an element-instance mask encoder.
+    has_element_instances: bool = False
 
 
 @dataclass

{code_loader-1.0.205.dev0 → code_loader-1.0.205.dev1}/code_loader/inner_leap_binder/leapbinder.py

@@ -1032,18 +1032,17 @@ class LeapBinder:
 
         Each entry must be one of:
           - "foreground" (the friendly name for the max-pooled default LS)
-          - "balanced" (the friendly name for the universal fallback default LS),
-            allowed only when a custom latent space is registered with
-            use_ls_for_analysis=True — that flagged LS is what replaces it
           - An input name registered via set_input(name=...)
           - A prediction name registered via add_prediction(name=...)
 
         Forbidden tokens (raise with a clear message):
+          - "balanced" — the friendly name for the universal fallback default
+            LS; cannot be disabled.
           - "user_custom" — populated by set_custom_latent_space; if the user
             registered a custom LS they should use it, not ignore it.
 
-        Order-independent: this runs after all set_input / add_prediction /
-        set_custom_latent_space calls because it consults setup_container.
+        Order-independent: this runs after all set_input / add_prediction calls
+        because it consults setup_container.inputs and prediction_types.
         """
         if not self.leap_analysis_configuration:
             return
@@ -1051,37 +1050,24 @@ class LeapBinder:
         if not names:
             return
 
-        forbidden = {"user_custom"}
+        forbidden = {"balanced", "user_custom"}
         input_names = {h.name for h in self.setup_container.inputs}
         prediction_names = {h.name for h in self.setup_container.prediction_types}
-        allowed_tokens = {"foreground", "balanced"} | input_names | prediction_names
-        analysis_ls_names = [
-            ls_name
-            for ls_name, handler in self.setup_container.custom_latent_spaces.items()
-            if handler.use_ls_for_analysis
-        ]
+        allowed_tokens = {"foreground"} | input_names | prediction_names
 
         for name in names:
             if name in forbidden:
                 raise Exception(
-                    f"Latent space '{name}' cannot be ignored. 'user_custom' is reserved "
-                    f"for set_custom_latent_space."
-                )
-            if name == "balanced" and not analysis_ls_names:
-                raise Exception(
-                    "Latent space 'balanced' (the model-derived default latent space) can "
-                    "only be ignored when a replacement is registered. Add "
-                    "@tensorleap_custom_latent_space(..., use_ls_for_analysis=True) to the "
-                    "latent space that should be analyzed instead, or drop 'balanced' from "
-                    "ignore_latent_spaces — ignoring it with nothing in its place would "
-                    "leave the analyses with no latent space."
+                    f"Latent space '{name}' cannot be ignored. "
+                    f"'balanced' is the universal fallback default LS and 'user_custom' is "
+                    f"reserved for set_custom_latent_space."
                 )
             if name not in allowed_tokens:
                 raise Exception(
                     f"Latent space name '{name}' in ignore_latent_spaces did not match "
                     f"any registered input or prediction. Available names: "
                     f"inputs={sorted(input_names)}, predictions={sorted(prediction_names)}, "
-                    f"tokens=['foreground', 'balanced']."
+                    f"tokens=['foreground']."
                 )
 
     def set_batch_size_to_validate(self, batch_size: int) -> None:

{code_loader-1.0.205.dev0 → code_loader-1.0.205.dev1}/code_loader/leaploader.py

@@ -935,7 +935,8 @@ class LeapLoader(LeapLoaderBase):
 
         return DatasetSetup(preprocess=dataset_preprocess, inputs=inputs, outputs=ground_truths,
                             metadata=metadata_instances, visualizers=visualizers, prediction_types=prediction_types,
-                            custom_losses=custom_losses, metrics=metrics, simulations=simulations)
+                            custom_losses=custom_losses, metrics=metrics, simulations=simulations,
+                            has_element_instances=bool(setup.instance_masks))
 
     def get_model_setup_response(self) -> ModelSetup:
         setup = global_leap_binder.setup_container

{code_loader-1.0.205.dev0 → code_loader-1.0.205.dev1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "1.0.205.dev0"
+version = "1.0.205.dev1"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"