code-loader 1.0.157.dev25__tar.gz → 1.0.158__tar.gz

{code_loader-1.0.157.dev25 → code_loader-1.0.158}/PKG-INFO

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

{code_loader-1.0.157.dev25 → code_loader-1.0.158}/code_loader/contract/datasetclasses.py

@@ -82,7 +82,7 @@ class PreprocessResponse:
 class ElementInstance:
     name: str
     mask: npt.NDArray[np.float32]
-    instance_metadata: Dict[str, Union[Optional[str], int, bool, Optional[float]]]
+    instance_metadata: Dict[str, Union[Optional[str], int, bool, Optional[float]]] = None
 
 SectionCallableInterface = Callable[[Union[int, str], PreprocessResponse], npt.NDArray[np.float32]]
 InstanceCallableInterface = Callable[[Union[int, str], PreprocessResponse, int], Optional[ElementInstance]]

{code_loader-1.0.157.dev25 → code_loader-1.0.158}/code_loader/leaploader.py

@@ -26,7 +26,8 @@ from code_loader.contract.responsedataclasses import DatasetIntegParseResult, Da
 from code_loader.inner_leap_binder import global_leap_binder
 from code_loader.inner_leap_binder.leapbinder import mapping_runtime_mode_env_var_mame
 from code_loader.leaploaderbase import LeapLoaderBase
-from code_loader.utils import get_root_exception_file_and_line_number
+from code_loader.utils import get_root_exception_file_and_line_number, get_metadata_type_from_variable
+
 
 class LeapLoader(LeapLoaderBase):
     def __init__(self, code_path: str, code_entry_name: str):
@@ -384,19 +385,8 @@ class LeapLoader(LeapLoaderBase):
                 metadata_instances.append(DatasetMetadataInstance(name=handler_test_payload.name,
                                                                   type=dataset_metadata_type))
                 continue
-            metadata_type = type(handler_test_payload.raw_result)
-            if metadata_type == int or isinstance(handler_test_payload.raw_result,
-                                                  (np.unsignedinteger, np.signedinteger)):
-                metadata_type = float
-            if isinstance(handler_test_payload.raw_result, str):
-                dataset_metadata_type = DatasetMetadataType.string
-            elif metadata_type == bool or isinstance(handler_test_payload.raw_result, np.bool_):
-                dataset_metadata_type = DatasetMetadataType.boolean
-            elif metadata_type == float or isinstance(handler_test_payload.raw_result, np.floating):
-                dataset_metadata_type = DatasetMetadataType.float
-            else:
-                raise Exception(f"Unsupported return type of metadata {handler_test_payload.name}."
-                                f"The return type should be one of [int, float, str, bool]. Got {metadata_type}")
+
+            dataset_metadata_type = get_metadata_type_from_variable(handler_test_payload.raw_result)
             metadata_instances.append(DatasetMetadataInstance(name=handler_test_payload.name,
                                                               type=dataset_metadata_type))
 

{code_loader-1.0.157.dev25 → code_loader-1.0.158}/pyproject.toml

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