code-loader 1.0.99.dev1__tar.gz → 1.0.99.dev2__tar.gz

{code_loader-1.0.99.dev1 → code_loader-1.0.99.dev2}/PKG-INFO

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

{code_loader-1.0.99.dev1 → code_loader-1.0.99.dev2}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -78,7 +78,7 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
             return ModelPlaceholder()
 
 
-        if os.environ[mapping_runtime_mode_env_var_mame]:
+        if os.environ.get(mapping_runtime_mode_env_var_mame):
             return mapping_inner
         else:
             return load_model_func
@@ -201,7 +201,7 @@ def tensorleap_custom_metric(name: str,
         mapping_inner.arg_names = leap_binder.setup_container.metrics[-1].metric_handler_data.arg_names
         mapping_inner.name = name
 
-        if os.environ[mapping_runtime_mode_env_var_mame]:
+        if os.environ.get(mapping_runtime_mode_env_var_mame):
             return mapping_inner
         else:
             return inner
@@ -279,7 +279,7 @@ def tensorleap_custom_visualizer(name: str, visualizer_type: LeapDataType,
         mapping_inner.arg_names = leap_binder.setup_container.visualizers[-1].visualizer_handler_data.arg_names
         mapping_inner.name = name
 
-        if os.environ[mapping_runtime_mode_env_var_mame]:
+        if os.environ.get(mapping_runtime_mode_env_var_mame):
             return mapping_inner
         else:
             return inner
@@ -447,7 +447,7 @@ def tensorleap_input_encoder(name: str, channel_dim=-1, model_input_index=None):
 
         mapping_inner.node_mapping = NodeMapping(name, node_mapping_type)
 
-        if os.environ[mapping_runtime_mode_env_var_mame]:
+        if os.environ.get(mapping_runtime_mode_env_var_mame):
             return mapping_inner
         else:
             return inner
@@ -502,7 +502,7 @@ def tensorleap_gt_encoder(name: str):
 
         mapping_inner.node_mapping = NodeMapping(name, NodeMappingType.GroundTruth)
 
-        if os.environ[mapping_runtime_mode_env_var_mame]:
+        if os.environ.get(mapping_runtime_mode_env_var_mame):
             return mapping_inner
         else:
             return inner
@@ -576,7 +576,7 @@ def tensorleap_custom_loss(name: str, connects_to=None):
         mapping_inner.arg_names = leap_binder.setup_container.custom_loss_handlers[-1].custom_loss_handler_data.arg_names
         mapping_inner.name = name
 
-        if os.environ[mapping_runtime_mode_env_var_mame]:
+        if os.environ.get(mapping_runtime_mode_env_var_mame):
             return mapping_inner
         else:
             return inner

{code_loader-1.0.99.dev1 → code_loader-1.0.99.dev2}/pyproject.toml

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