code-loader 1.0.136__py3-none-any.whl → 1.0.137__py3-none-any.whl

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -2,6 +2,7 @@
 import os
 from collections import defaultdict
 from functools import lru_cache
+from pathlib import Path
 from typing import Optional, Union, Callable, List, Dict
 
 import numpy as np
@@ -12,7 +13,7 @@ from code_loader.contract.datasetclasses import CustomCallableInterfaceMultiArgs
     VisualizerCallableInterface, MetadataSectionCallableInterface, PreprocessResponse, SectionCallableInterface, \
     ConfusionMatrixElement, SamplePreprocessResponse, PredictionTypeHandler, InstanceCallableInterface, ElementInstance, \
     InstanceLengthCallableInterface
-from code_loader.contract.enums import MetricDirection, LeapDataType, DatasetMetadataType
+from code_loader.contract.enums import MetricDirection, LeapDataType, DatasetMetadataType, DataStateType
 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, \
@@ -44,7 +45,7 @@ def _add_mapping_connections(connects_to, arg_names, node_mapping_type, name):
         _add_mapping_connection(user_unique_name, connection_destinations, arg_names, name, node_mapping_type)
 
 
-def integration_test():
+def tensorleap_integration_test():
     def decorating_function(integration_test_function: Callable):
         leap_binder.integration_test_func = integration_test_function
 
@@ -56,10 +57,19 @@ def integration_test():
 
                 try:
                     os.environ[mapping_runtime_mode_env_var_mame] = 'True'
-                    integration_test_function(None, None)
+                    integration_test_function(None, PreprocessResponse(state=DataStateType.training, length=0))
                 except Exception as e:
-                    print(f'Error during integration test: Make sure to disable any non tensorleap decorators '
-                          f'functions before pushing a new TL version')
+                    import traceback
+                    first_tb = traceback.extract_tb(e.__traceback__)[-1]
+                    file_name = Path(first_tb.filename).name
+                    line_number = first_tb.lineno
+                    if isinstance(e, TypeError) and 'is not subscriptable' in str(e):
+                        print(f'Invalid integration code. File {file_name}, line {line_number}: '
+                              f'Please remove this indexing operation usage from the integration test code.')
+                    else:
+                        print(f'Invalid integration code. File {file_name}, line {line_number}: '
+                              f'Integration test is only allowed to call Tensorleap decorators. '
+                              f'Ensure any arithmetics, external library use, Python logic is placed within Tensorleap decoders')
                 finally:
                     if mapping_runtime_mode_env_var_mame in os.environ:
                         del os.environ[mapping_runtime_mode_env_var_mame]
@@ -78,7 +88,7 @@ def _safe_get_item(key):
     except ValueError:
         raise Exception(f'Tensorleap currently supports models with no more then 10 inputs')
 
-def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]] = None):
+def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]] = []):
     for i, prediction_type in enumerate(prediction_types):
         leap_binder.add_prediction(prediction_type.name, prediction_type.labels, prediction_type.channel_dim, i)
 

code_loader/leaploader.py

@@ -51,7 +51,7 @@ class LeapLoader(LeapLoaderBase):
             os.environ[mapping_runtime_mode_env_var_mame] = 'TRUE'
             self.evaluate_module()
             if global_leap_binder.integration_test_func is not None:
-                global_leap_binder.integration_test_func(None, None)
+                global_leap_binder.integration_test_func(None, PreprocessResponse(state=DataStateType.training, length=0))
         except TypeError as e:
             import traceback
             if "leap_binder.set_metadata(" in traceback.format_exc(5):

{code_loader-1.0.136.dist-info → code_loader-1.0.137.dist-info}/METADATA

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

{code_loader-1.0.136.dist-info → code_loader-1.0.137.dist-info}/RECORD

@@ -21,8 +21,8 @@ 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=EptQhu4dPVXmAfUnv_Whzmess0ueRYy5RLpm7WYe6Ho,50981
-code_loader/leaploader.py,sha256=85XUWd7Y8kup76xeqefgd8db7pGaeHMhE7QlmYlNFMw,29747
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=7lykS1VoaO58HIf_ysegvFxYDVI8x3AYMMa_ODz2fXA,51767
+code_loader/leaploader.py,sha256=PC7d36_v_7RWO67uOVffsmAllT-KXCJAqpMYM5GjHoU,29801
 code_loader/leaploaderbase.py,sha256=LIFcC6xo6V_iiGN3BjibXETu_l84EWM_WIOKAvkfTiM,4458
 code_loader/mixpanel_tracker.py,sha256=l9z_szKKQ7apEbdNZpGH1TKAiT_TsBHb9AQnePaWTyo,4942
 code_loader/plot_functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -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.136.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.136.dist-info/METADATA,sha256=pNpIswJBMtQtorkL80niuBszYQ38F9repsK8WOCHRJ8,1090
-code_loader-1.0.136.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.136.dist-info/RECORD,,
+code_loader-1.0.137.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.137.dist-info/METADATA,sha256=NSUMwxhMsdTdfuD5zuztuUa8ZcBzlOhdPhhd0CuMjJw,1090
+code_loader-1.0.137.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.137.dist-info/RECORD,,