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

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

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

{code_loader-1.0.93.dev2 → code_loader-1.0.93.dev4}/code_loader/contract/datasetclasses.py

@@ -205,7 +205,7 @@ class MetadataHandler:
 class PredictionTypeHandler:
     name: str
     labels: List[str]
-    channel_dim: int
+    channel_dim: int = -1
 
 
 @dataclass

{code_loader-1.0.93.dev2 → code_loader-1.0.93.dev4}/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, PredictionTypeHandler
 from code_loader.contract.enums import MetricDirection, LeapDataType, DatasetMetadataType
 from code_loader import leap_binder
 from code_loader.contract.mapping import NodeMapping, NodeMappingType, NodeConnection
@@ -16,7 +16,7 @@ from code_loader.contract.visualizer_classes import LeapImage, LeapImageMask, Le
     LeapHorizontalBar, LeapImageWithBBox, LeapImageWithHeatmap
 
 
-mapping_runtime_mode_env_var_mame = '__mapping_runtime_mode__'
+mapping_runtime_mode_env_var_mame = '__MAPPING_RUNTIME_MODE__'
 
 
 def _add_mapping_connection(user_unique_name, connection_destinations, arg_names, name, node_mapping_type):
@@ -35,7 +35,10 @@ def _add_mapping_connections(connects_to, arg_names, node_mapping_type, name):
 
 
 
-def tensorleap_load_model():
+def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]] = None):
+    for i, prediction_type in enumerate(prediction_types):
+        leap_binder.add_prediction(prediction_type.name, prediction_type.labels, prediction_type.channel_dim, i)
+
     def decorating_function(load_model_func):
         class TempMapping:
             pass

{code_loader-1.0.93.dev2 → code_loader-1.0.93.dev4}/code_loader/leaploader.py

@@ -2,6 +2,7 @@
 import importlib.util
 import inspect
 import io
+import os
 import sys
 from contextlib import redirect_stdout
 from functools import lru_cache
@@ -22,6 +23,7 @@ from code_loader.contract.responsedataclasses import DatasetIntegParseResult, Da
     VisualizerInstance, PredictionTypeInstance, ModelSetup, CustomLayerInstance, MetricInstance, CustomLossInstance, \
     EngineFileContract
 from code_loader.inner_leap_binder import global_leap_binder
+from code_loader.inner_leap_binder.leapbinder_decorators 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
 
@@ -157,6 +159,7 @@ class LeapLoader(LeapLoaderBase):
         stdout_steam = io.StringIO()
         with redirect_stdout(stdout_steam):
             try:
+                os.environ[mapping_runtime_mode_env_var_mame] = 'TRUE'
                 self.exec_script()
                 preprocess_test_payload = self._check_preprocess()
                 test_payloads.append(preprocess_test_payload)
@@ -173,6 +176,10 @@ class LeapLoader(LeapLoaderBase):
                 general_error = f"Something went wrong. {repr(e.__cause__)} in file {file_name}, line_number:  {line_number}\nStacktrace:\n{stacktrace}"
                 is_valid = False
 
+            del os.environ[mapping_runtime_mode_env_var_mame]
+
+
+
         print_log = stdout_steam.getvalue()
         is_valid_for_model = bool(global_leap_binder.setup_container.custom_layers)
         model_setup = self.get_model_setup_response()

{code_loader-1.0.93.dev2 → code_loader-1.0.93.dev4}/pyproject.toml

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