code-loader 1.0.72.dev3__py3-none-any.whl → 1.0.72.dev4__py3-none-any.whl

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.dist-info → code_loader-1.0.72.dev4.dist-info}/METADATA

@@ -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.dist-info → code_loader-1.0.72.dev4.dist-info}/RECORD

@@ -22,11 +22,11 @@ code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_p
 code_loader/inner_leap_binder/leapbinder.py,sha256=-fryKzD8T8K2EgrOsR5NryabP8_1k_m3POLwhYIA_8I,26708
 code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=ebMxknpKMW-dE8Erq0fFq4RrE5E_Jfx9IvmRRZSdhlc,20813
 code_loader/leaploader.py,sha256=g160Z0MRop_7m3bu1HawcGoZNKakvs9Su4tEfbQ9pR0,22914
-code_loader/leaploaderbase.py,sha256=FDXjTFBjnpfUyZ2tS3wLo7y2j82Qb_VKHI1BMB5gFww,3269
+code_loader/leaploaderbase.py,sha256=ijTodEBL-Q9DulR9z0xU0fo72rVvm06VvoRyrxzoCtE,4012
 code_loader/utils.py,sha256=aw2i_fqW_ADjLB66FWZd9DfpCQ7mPdMyauROC5Nd51I,2197
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=Ffx5VHVOe5ujBOsjBSxN_aIEVwFSQ6gbhTMG5aUS-po,2305
-code_loader-1.0.72.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.72.dev3.dist-info/METADATA,sha256=-9q2S44uQTVZhb3uYyUp_T3ByP5WuljvF9blRuuly3g,854
-code_loader-1.0.72.dev3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.72.dev3.dist-info/RECORD,,
+code_loader-1.0.72.dev4.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.72.dev4.dist-info/METADATA,sha256=TQ8fEw3bxJr1ZLchIY_e8EMfRVWreauzQhnK426Ca2k,854
+code_loader-1.0.72.dev4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.72.dev4.dist-info/RECORD,,