code-loader 1.0.139.dev2__py3-none-any.whl → 1.0.139.dev3__py3-none-any.whl

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -174,7 +174,8 @@ def tensorleap_integration_test():
             global _called_from_inside_tl_integration_test_decorator
             try:
                 _called_from_inside_tl_integration_test_decorator = True
-                _update_env_status("tensorleap_integration_test", "v")#put here because otherwise it will become  v only if it finishes all the script
+                if not _update_env_status is None:
+                    _update_env_status("tensorleap_integration_test", "v")#put here because otherwise it will become  v only if it finishes all the script
                 ret = integration_test_function(*args, **kwargs)
 
                 try:
@@ -320,7 +321,8 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
                 def get_inputs(self):
                     return self.model.get_inputs()
             model_placeholder=ModelPlaceholder()
-            _update_env_status("tensorleap_load_model", "v")
+            if not _update_env_status is None:
+                _update_env_status("tensorleap_load_model", "v")
             return model_placeholder
 
         def mapping_inner():
@@ -600,7 +602,8 @@ def tensorleap_custom_metric(name: str,
             result = inner_without_validate(*args, **kwargs)
 
             _validate_result(result)
-            _update_env_status("tensorleap_custom_metric","v")
+            if not _update_env_status is None:
+                _update_env_status("tensorleap_custom_metric","v")
             return result
 
         def mapping_inner(*args, **kwargs):
@@ -721,7 +724,8 @@ def tensorleap_custom_visualizer(name: str, visualizer_type: LeapDataType,
             result = inner_without_validate(*args, **kwargs)
 
             _validate_result(result)
-            _update_env_status("tensorleap_custom_visualizer","v")
+            if not _update_env_status is None:
+                _update_env_status("tensorleap_custom_visualizer","v")
             return result
 
         def mapping_inner(*args, **kwargs):
@@ -810,7 +814,8 @@ def tensorleap_metadata(
             result = inner_without_validate(sample_id, preprocess_response)
 
             _validate_result(result)
-            _update_env_status("tensorleap_metadata","v")
+            if not _update_env_status is None:
+                _update_env_status("tensorleap_metadata","v")
             return result
 
         return inner
@@ -899,7 +904,8 @@ def tensorleap_preprocess():
             _validate_input_args(*args, **kwargs)
             result = user_function()
             _validate_result(result)
-            _update_env_status("tensorleap_preprocess", "v")
+            if not _update_env_status is None:
+                _update_env_status("tensorleap_preprocess", "v")
             return result
 
         return inner
@@ -1141,7 +1147,8 @@ def tensorleap_input_encoder(name: str, channel_dim=-1, model_input_index=None):
             if _called_from_inside_tl_decorator == 0 and _called_from_inside_tl_integration_test_decorator:
                 batch_warning(result,user_function.__name__)
                 result = np.expand_dims(result, axis=0)
-            _update_env_status("tensorleap_input_encoder", "v")
+            if not _update_env_status is None:
+                _update_env_status("tensorleap_input_encoder", "v")
 
             return result
 
@@ -1225,8 +1232,8 @@ def tensorleap_gt_encoder(name: str):
             if _called_from_inside_tl_decorator == 0 and _called_from_inside_tl_integration_test_decorator:
                 batch_warning(result, user_function.__name__)
                 result = np.expand_dims(result, axis=0)
-            _update_env_status("tensorleap_gt_encoder", "v")
-
+            if not _update_env_status is None:
+                _update_env_status("tensorleap_gt_encoder", "v")
             return result
 
         inner.node_mapping = NodeMapping(name, NodeMappingType.GroundTruth)
@@ -1328,7 +1335,8 @@ def tensorleap_custom_loss(name: str, connects_to=None):
             result = inner_without_validate(*args, **kwargs)
 
             _validate_result(result)
-            _update_env_status("tensorleap_custom_loss", "v")
+            if not _update_env_status is None:
+                _update_env_status("tensorleap_custom_loss", "v")
 
             return result
 

{code_loader-1.0.139.dev2.dist-info → code_loader-1.0.139.dev3.dist-info}/METADATA

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

{code_loader-1.0.139.dev2.dist-info → code_loader-1.0.139.dev3.dist-info}/RECORD

@@ -21,7 +21,7 @@ code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZ
 code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaSbeTVzq-2ja_SQw4zi7LXwKL9cY,990
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
 code_loader/inner_leap_binder/leapbinder.py,sha256=Q3D9yVM-GNEJfYRFvMV__BoZbcWOgnWKhrZXAv6Tu7o,33232
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=NlH8W1q8kVAik81EJ9ZNjWkAm8XPSW-XrKh9wBgd5BA,69593
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=Rgt_9Kllc0yZVXPsk1yC1p2Yb2OMx3QXMQlQvuixpZ0,70055
 code_loader/leaploader.py,sha256=6D6xZzMI6qSNIb3tuKLB3BbK5H8QS1_r7iQjIXO3OkM,29795
 code_loader/leaploaderbase.py,sha256=LIFcC6xo6V_iiGN3BjibXETu_l84EWM_WIOKAvkfTiM,4458
 code_loader/mixpanel_tracker.py,sha256=l9z_szKKQ7apEbdNZpGH1TKAiT_TsBHb9AQnePaWTyo,4942
@@ -31,7 +31,7 @@ code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6Lz
 code_loader/utils.py,sha256=gXENTYpjdidq2dx0gVbXlErPeHoNs-4TYAZbLRe0y2c,2712
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
-code_loader-1.0.139.dev2.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.139.dev2.dist-info/METADATA,sha256=bZ4R5F5wQdWZEg666Lxiw1LxHcqUCmRXsgKvxFUGof4,1095
-code_loader-1.0.139.dev2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-code_loader-1.0.139.dev2.dist-info/RECORD,,
+code_loader-1.0.139.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.139.dev3.dist-info/METADATA,sha256=MQ-BRAy9_HLMKQnO0Dv0dHoZ8NGwbfd90eZULo_hcx4,1095
+code_loader-1.0.139.dev3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+code_loader-1.0.139.dev3.dist-info/RECORD,,