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

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

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

{code_loader-1.0.61.dev1 → code_loader-1.0.61.dev2}/code_loader/leaploaderbase.py

@@ -8,7 +8,7 @@ import numpy as np
 import numpy.typing as npt
 
 from code_loader.contract.datasetclasses import DatasetSample, VisualizerHandler, LeapData, CustomLossHandler, \
-    PredictionTypeHandler, CustomLayerHandler
+    PredictionTypeHandler, CustomLayerHandler, MetricHandler
 from code_loader.contract.enums import DataStateEnum
 from code_loader.contract.responsedataclasses import DatasetIntegParseResult, DatasetTestResultPayload, \
     DatasetSetup, ModelSetup
@@ -19,10 +19,8 @@ class LeapLoaderBase:
         self.code_entry_name = code_entry_name
         self.code_path = code_path
 
-        self._preprocess_result_cached = None
-
     @abstractmethod
-    def metric_by_name(self):
+    def metric_by_name(self) -> Dict[str, MetricHandler]:
         pass
 
     @abstractmethod
@@ -75,3 +73,9 @@ class LeapLoaderBase:
     @abstractmethod
     def get_sample_id_type(self) -> Type:
         pass
+
+    def is_custom_latent_space(self) -> bool:
+        if not self.code_entry_name or not self.code_path:
+            return False
+        custom_layers = self.custom_layers()
+        return any(layer.use_custom_latent_space for layer in custom_layers.values())

{code_loader-1.0.61.dev1 → code_loader-1.0.61.dev2}/pyproject.toml

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