code-loader 1.0.208.dev1__tar.gz → 1.0.208.dev2__tar.gz

{code_loader-1.0.208.dev1 → code_loader-1.0.208.dev2}/PKG-INFO

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

{code_loader-1.0.208.dev1 → code_loader-1.0.208.dev2}/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-1.0.208.dev1 → code_loader-1.0.208.dev2}/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 → code_loader-1.0.208.dev2}/pyproject.toml

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