code-loader 1.0.208.dev1__py3-none-any.whl → 1.0.208.dev3__py3-none-any.whl

code_loader/inner_leap_binder/leapbinder.py

@@ -586,8 +586,7 @@ class LeapBinder:
             space for the Out-Of-Distribution and Domain-Gap insights instead of the
             built-in defaults. At most one registered custom latent space may set this;
             registering a second one with the flag raises. Not currently supported when
-            instance_aware=True, or for a model-computed latent space (computed_at='model')
-            — it is ignored (with a warning) and forced to False.
+            instance_aware=True — it is ignored (with a warning) and forced to False.
         instance_aware (bool): When True, `function` takes a third `instance_id` argument
             and is called once per element-instance row instead of once per sample.
         """
@@ -616,16 +615,6 @@ class LeapBinder:
                 f"latent space ('{name}'). Ignoring it; the flag will be set to False."
             )
             use_ls_for_analysis = False
-        # Model-computed latent spaces are fetched via run_model_latent_space, a separate path
-        # that get_sample's per-sample custom_latent_spaces dict never populates — the engine has
-        # no consumer for a model-computed analysis latent space yet, so force it off rather than
-        # silently accepting a flag that has no effect.
-        if computed_at == 'model' and use_ls_for_analysis:
-            warnings.warn(
-                f"use_ls_for_analysis=True is not currently supported for a model-computed custom "
-                f"latent space ('{name}'). Ignoring it; the flag will be set to False."
-            )
-            use_ls_for_analysis = False
         if use_ls_for_analysis:
             already_flagged = [
                 existing_name

code_loader/leaploader.py

@@ -1470,19 +1470,24 @@ class LeapLoader(LeapLoaderBase):
                      if handler.computed_at == 'model')
 
     @lru_cache()
-    def get_model_latent_space_specs(self) -> Dict[str, Dict[str, Any]]:
+    def get_custom_latent_space_specs(self) -> Dict[str, Dict[str, Any]]:
         self.exec_script()
         return {
             name: {
+                'computed_at': handler.computed_at,
                 'arg_names': list(handler.arg_names or []),
                 'reduce': handler.reduce.value if handler.reduce is not None else None,
                 'n_components': handler.n_components,
                 'channel_axis': handler.channel_axis,
             }
             for name, handler in global_leap_binder.setup_container.custom_latent_spaces.items()
-            if handler.computed_at == 'model'
         }
 
+    @lru_cache()
+    def get_model_latent_space_specs(self) -> Dict[str, Dict[str, Any]]:
+        return {name: spec for name, spec in self.get_custom_latent_space_specs().items()
+                if spec['computed_at'] == 'model'}
+
     def run_model_latent_space(self, ls_name: str, sample_ids: np.array, state: DataStateEnum,
                                input_tensors_by_arg_name: Dict[str, npt.NDArray[np.float32]]
                                ) -> npt.NDArray[np.float32]:

code_loader/leaploaderbase.py

@@ -239,6 +239,11 @@ class LeapLoaderBase:
         raise NotImplementedError(f'{type(self).__name__} does not implement '
                                   'get_model_custom_latent_space_names.')
 
+    @abstractmethod
+    def get_custom_latent_space_specs(self) -> Dict[str, Dict[str, Any]]:
+        raise NotImplementedError(f'{type(self).__name__} does not implement '
+                                  'get_custom_latent_space_specs.')
+
     @abstractmethod
     def get_model_latent_space_specs(self) -> Dict[str, Dict[str, Any]]:
         raise NotImplementedError(f'{type(self).__name__} does not implement '

{code_loader-1.0.208.dev1.dist-info → code_loader-1.0.208.dev3.dist-info}/METADATA

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

{code_loader-1.0.208.dev1.dist-info → code_loader-1.0.208.dev3.dist-info}/RECORD

@@ -21,10 +21,10 @@ code_loader/experiment_api/types.py,sha256=MY8xFARHwdVA7p4dxyhD60ShmttgTvb4qdp1o
 code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZJEqBqc,3304
 code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaSbeTVzq-2ja_SQw4zi7LXwKL9cY,990
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
-code_loader/inner_leap_binder/leapbinder.py,sha256=J5rWu8caFNkH3I4J9jkIoWimV188jhI1XBV6-b1thmc,63883
+code_loader/inner_leap_binder/leapbinder.py,sha256=j6oa6oyM04scUCmyFKB80E0MTYDBDUcF9cwiyJTj5oQ,63134
 code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=fAoXt1N9zUI6wqCk5NyuPQJBhO0ZtOFIyM36OiA5XJ8,206593
-code_loader/leaploader.py,sha256=2KzjvMgXia-yhpOXIg2pwEOs2QgM1ZALXppXcuyP1N8,95876
-code_loader/leaploaderbase.py,sha256=ok7-vjbaNX_zeW8UUEUvQJSce5yB7qF4-uNLSxr0dGI,12359
+code_loader/leaploader.py,sha256=xC7oT_2VV2VhhCzBb1J27nQrRAGcJwq04eypatENOz8,96115
+code_loader/leaploaderbase.py,sha256=cQwlJdx_tLJXMxTq8AURsFpyF0s3jpqkWaF5i1r8hSE,12601
 code_loader/mixpanel_tracker.py,sha256=rNwRmFifNbdUoqLQvvhhgpKczWpWiEmd8MfyJe27sxw,9131
 code_loader/plot_functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/plot_functions/plot_functions.py,sha256=2DC-zlVaN13P4VNx5d8csgs80C6SisaeP1-Kq2LW7iM,16075
@@ -32,7 +32,7 @@ code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6Lz
 code_loader/utils.py,sha256=Auw39lKpEQ7Ld2q1nj6S18GIDKWcjCxU0TDrur2UEko,11968
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=grTPin_lCE9aci8i8CqA7DqQwAyXRB7_EamA3na_pls,5438
-code_loader-1.0.208.dev1.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.208.dev1.dist-info/METADATA,sha256=Bkb4Z-t1BOLRLX-GctwTpwfhHi6NHqRAEvXQLuDPu_8,1095
-code_loader-1.0.208.dev1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.208.dev1.dist-info/RECORD,,
+code_loader-1.0.208.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.208.dev3.dist-info/METADATA,sha256=_PM6rMpE1DSLWHREFJ-sKGfRAWJWoI_tet4gMNj3rEo,1095
+code_loader-1.0.208.dev3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.208.dev3.dist-info/RECORD,,