code-loader 1.0.91.dev5__tar.gz → 1.0.91.dev6__tar.gz

{code_loader-1.0.91.dev5 → code_loader-1.0.91.dev6}/PKG-INFO

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

{code_loader-1.0.91.dev5 → code_loader-1.0.91.dev6}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -8,7 +8,7 @@ import numpy.typing as npt
 from code_loader.contract.datasetclasses import CustomCallableInterfaceMultiArgs, \
     CustomMultipleReturnCallableInterfaceMultiArgs, ConfusionMatrixCallableInterfaceMultiArgs, CustomCallableInterface, \
     VisualizerCallableInterface, MetadataSectionCallableInterface, PreprocessResponse, SectionCallableInterface, \
-    ConfusionMatrixElement, SamplePreprocessResponse
+    ConfusionMatrixElement, SamplePreprocessResponse, InstanceCallableInterface
 from code_loader.contract.enums import MetricDirection, LeapDataType, DatasetMetadataType
 from code_loader import leap_binder
 from code_loader.contract.mapping import NodeMapping, NodeMappingType, NodeConnection
@@ -347,23 +347,30 @@ def tensorleap_unlabeled_preprocess():
     return decorating_function
 
 
-def tensorleap_instance_encoder():
-    def decorating_function(user_function: Callable[[], PreprocessResponse]):
-        leap_binder.set_unlabeled_data_preprocess(user_function)
+def tensorleap_instances_masks_encoder(name: str, model_input_index=None):
+    def decorating_function(user_function: InstanceCallableInterface):
+        leap_binder.set_instance_masks(user_function, name)
 
-        def _validate_input_args(*args, **kwargs):
-            assert len(args) == 0 and len(kwargs) == 0, \
-                (f'tensorleap_instance_encoder validation failed: '
-                 f'The function should not take any arguments. Got {args} and {kwargs}.')
+        def _validate_input_args(sample_id: Union[int, str], preprocess_response: PreprocessResponse):
+            assert isinstance(sample_id, (int, str)), \
+                (f'tensorleap_input_encoder validation failed: '
+                 f'Argument sample_id should be either int or str. Got {type(sample_id)}.')
+            assert isinstance(preprocess_response, PreprocessResponse), \
+                (f'tensorleap_input_encoder validation failed: '
+                 f'Argument preprocess_response should be a PreprocessResponse. Got {type(preprocess_response)}.')
+            assert type(sample_id) == preprocess_response.sample_id_type, \
+                (f'tensorleap_input_encoder validation failed: '
+                 f'Argument sample_id should be as the same type as defined in the preprocess response '
+                 f'{preprocess_response.sample_id_type}. Got {type(sample_id)}.')
 
         def _validate_result(result):
-            assert isinstance(result, PreprocessResponse), \
-                (f'tensorleap_instance_encoder validation failed: '
-                 f'The return type should be a PreprocessResponse. Got {type(result)}.')
+            assert isinstance(result, list), \
+                (f'tensorleap_input_encoder validation failed: '
+                 f'Unsupported return type. Should be a numpy array. Got {type(result)}.')
 
-        def inner(*args, **kwargs):
-            _validate_input_args(*args, **kwargs)
-            result = user_function()
+        def inner(sample_id, preprocess_response):
+            _validate_input_args(sample_id, preprocess_response)
+            result = user_function(sample_id, preprocess_response)
             _validate_result(result)
             return result
 

{code_loader-1.0.91.dev5 → code_loader-1.0.91.dev6}/code_loader/utils.py

@@ -5,8 +5,6 @@ from typing import List, Union, Tuple, Any, Callable
 import traceback
 import numpy as np
 import numpy.typing as npt
-from numpy import ndarray, dtype, floating
-from numpy._typing import _32Bit
 
 from code_loader.contract.datasetclasses import SectionCallableInterface, PreprocessResponse, \
     InstanceCallableInterface

{code_loader-1.0.91.dev5 → code_loader-1.0.91.dev6}/pyproject.toml

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