code-loader 1.0.93.dev3__tar.gz → 1.0.93.dev5__tar.gz

{code_loader-1.0.93.dev3 → code_loader-1.0.93.dev5}/PKG-INFO

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

{code_loader-1.0.93.dev3 → code_loader-1.0.93.dev5}/code_loader/inner_leap_binder/leapbinder.py

@@ -24,6 +24,9 @@ from code_loader.visualizers.default_visualizers import DefaultVisualizer, \
     default_image_mask_visualizer, default_text_mask_visualizer, default_raw_data_visualizer, default_video_visualizer
 
 
+mapping_runtime_mode_env_var_mame = '__MAPPING_RUNTIME_MODE__'
+
+
 class LeapBinder:
     """
     Interface to the Tensorleap platform. Provides methods to set up preprocessing,

{code_loader-1.0.93.dev3 → code_loader-1.0.93.dev5}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -14,9 +14,7 @@ from code_loader import leap_binder
 from code_loader.contract.mapping import NodeMapping, NodeMappingType, NodeConnection
 from code_loader.contract.visualizer_classes import LeapImage, LeapImageMask, LeapTextMask, LeapText, LeapGraph, \
     LeapHorizontalBar, LeapImageWithBBox, LeapImageWithHeatmap
-
-
-mapping_runtime_mode_env_var_mame = '__mapping_runtime_mode__'
+from code_loader.inner_leap_binder.leapbinder import mapping_runtime_mode_env_var_mame
 
 
 def _add_mapping_connection(user_unique_name, connection_destinations, arg_names, name, node_mapping_type):

{code_loader-1.0.93.dev3 → code_loader-1.0.93.dev5}/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 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.dev3 → code_loader-1.0.93.dev5}/pyproject.toml

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