code-loader 1.0.72.dev3__tar.gz → 1.0.72.dev4__tar.gz

{code_loader-1.0.72.dev3 → code_loader-1.0.72.dev4}/PKG-INFO

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

{code_loader-1.0.72.dev3 → code_loader-1.0.72.dev4}/code_loader/leaploaderbase.py

@@ -10,7 +10,7 @@ import numpy.typing as npt
 from code_loader.contract.datasetclasses import DatasetSample, LeapData, \
     PredictionTypeHandler, CustomLayerHandler, VisualizerHandlerData, MetricHandlerData, MetricCallableReturnType, \
     CustomLossHandlerData
-from code_loader.contract.enums import DataStateEnum
+from code_loader.contract.enums import DataStateEnum, DataStateType
 from code_loader.contract.responsedataclasses import DatasetIntegParseResult, DatasetTestResultPayload, \
     DatasetSetup, ModelSetup
 
@@ -23,6 +23,23 @@ class LeapLoaderBase:
         self.current_working_sample_ids: Optional[np.array] = None
         self.current_working_state: Optional[DataStateEnum] = None
 
+    def set_current_working_sample_ids(self, sample_ids: np.array):
+        if type(sample_ids[0]) is bytes:
+            sample_ids = np.array([sample_id.decode('utf-8') for sample_id in sample_ids])
+        self.current_working_sample_ids = sample_ids
+
+    def set_current_working_state(self, state: Union[DataStateEnum, DataStateType, str, int, bytes]):
+        if type(state) is bytes:
+            state = DataStateEnum[state.decode('utf-8')]
+        elif type(state) is str:
+            state = DataStateEnum[state]
+        elif type(state) is int:
+            state = DataStateEnum(state)
+        elif type(state) is DataStateType:
+            state = DataStateEnum[state.name]
+
+        self.current_working_state = state
+
     @abstractmethod
     def metric_by_name(self) -> Dict[str, MetricHandlerData]:
         pass

{code_loader-1.0.72.dev3 → code_loader-1.0.72.dev4}/pyproject.toml

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