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

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

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

{code_loader-1.0.72.dev2 → code_loader-1.0.72.dev4}/code_loader/inner_leap_binder/leapbinder.py

@@ -240,15 +240,14 @@ class LeapBinder:
             leap_binder.add_custom_loss(custom_loss_function, name='custom_loss')
         """
 
-        regular_arg_names = []
+        regular_arg_names = inspect.getfullargspec(function)[0]
         preprocess_response_arg_name = None
         for arg_name, arg_type in inspect.getfullargspec(function).annotations.items():
             if arg_type == SamplePreprocessResponse:
                 if preprocess_response_arg_name is not None:
                     raise Exception("only one argument can be of type SamplePreprocessResponse")
                 preprocess_response_arg_name = arg_name
-            else:
-                regular_arg_names.append(arg_name)
+                regular_arg_names.remove(arg_name)
 
         self.setup_container.custom_loss_handlers.append(
             CustomLossHandler(CustomLossHandlerData(name, regular_arg_names), function))

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

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