code-loader 1.0.60.dev2__py3-none-any.whl → 1.0.62__py3-none-any.whl

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -58,11 +58,13 @@ def tensorleap_custom_metric(name: str, direction: Optional[MetricDirection] = M
                         assert isinstance(single_metric_result[0][0], ConfusionMatrixElement), \
                             f'{supported_types_message}Got List[List[{type(single_metric_result[0][0])}]].'
                     else:
-                        assert isinstance(single_metric_result[0], float), f'{supported_types_message}Got List[{type(single_metric_result[0])}].'
+                        assert isinstance(single_metric_result[0],
+                                          float), f'{supported_types_message}Got List[{type(single_metric_result[0])}].'
                 else:
-                    assert isinstance(single_metric_result, np.ndarray), f'{supported_types_message}Got {type(single_metric_result)}.'
+                    assert isinstance(single_metric_result,
+                                      np.ndarray), f'{supported_types_message}Got {type(single_metric_result)}.'
                     assert len(single_metric_result.shape) == 1, (f'tensorleap_custom_metric validation failed: '
-                                                    f'The return shape should be 1D. Got {len(single_metric_result.shape)}D.')
+                                                                  f'The return shape should be 1D. Got {len(single_metric_result.shape)}D.')
 
                 if leap_binder.batch_size_to_validate:
                     assert len(single_metric_result) == leap_binder.batch_size_to_validate, \
@@ -251,7 +253,7 @@ def tensorleap_input_encoder(name: str, channel_dim=-1):
             if input_handler.name == name:
                 raise Exception(f'Input with name {name} already exists. '
                                 f'Please choose another')
-        if channel_dim < 0 and channel_dim != -1:
+        if channel_dim <= 0 and channel_dim != -1:
             raise Exception(f"Channel dim for input {name} is expected to be either -1 or positive")
 
         leap_binder.set_input(user_function, name, channel_dim=channel_dim)
@@ -276,7 +278,7 @@ def tensorleap_input_encoder(name: str, channel_dim=-1):
                 (f'tensorleap_input_encoder validation failed: '
                  f'The return type should be a numpy array of type float32. Got {result.dtype}.')
             assert channel_dim - 1 <= len(result.shape), (f'tensorleap_input_encoder validation failed: '
-                 f'The channel_dim ({channel_dim}) should be <= to the rank of the resulting input rank ({len(result.shape)}).')
+                                                          f'The channel_dim ({channel_dim}) should be <= to the rank of the resulting input rank ({len(result.shape)}).')
 
         def inner(sample_id, preprocess_response):
             _validate_input_args(sample_id, preprocess_response)
@@ -345,11 +347,21 @@ def tensorleap_custom_loss(name: str):
                 raise Exception('the input arguments of the custom loss function should be tensorflow tensors') from e
 
             for i, arg in enumerate(args):
-                assert isinstance(arg, tf.Tensor), (f'tensorleap_custom_loss validation failed: '
-                                                    f'Argument #{i} should be a tensorflow tensor. Got {type(arg)}.')
+                if isinstance(arg, list):
+                    for y, elem in enumerate(arg):
+                        assert isinstance(elem, tf.Tensor), (f'tensorleap_custom_loss validation failed: '
+                                                             f'Element #{y} of list should be a tensorflow tensor. Got {type(elem)}.')
+                else:
+                    assert isinstance(arg, tf.Tensor), (f'tensorleap_custom_loss validation failed: '
+                                                        f'Argument #{i} should be a tensorflow tensor. Got {type(arg)}.')
             for _arg_name, arg in kwargs.items():
-                assert isinstance(arg, tf.Tensor), (f'tensorleap_custom_loss validation failed: '
-                                                    f'Argument {_arg_name} should be a tensorflow tensor. Got {type(arg)}.')
+                if isinstance(arg, list):
+                    for y, elem in enumerate(arg):
+                        assert isinstance(elem, tf.Tensor), (f'tensorleap_custom_loss validation failed: '
+                                                             f'Element #{y} of list should be a tensorflow tensor. Got {type(elem)}.')
+                else:
+                    assert isinstance(arg, tf.Tensor), (f'tensorleap_custom_loss validation failed: '
+                                                        f'Argument #{_arg_name} should be a tensorflow tensor. Got {type(arg)}.')
 
         def _validate_result(result):
             try:

{code_loader-1.0.60.dev2.dist-info → code_loader-1.0.62.dist-info}/METADATA

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

{code_loader-1.0.60.dev2.dist-info → code_loader-1.0.62.dist-info}/RECORD

@@ -18,12 +18,12 @@ 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=LVzpynjISO-a774flzGt1yAQPsSYNE8B5V58Hacs7bQ,25216
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=6yHYZRFb5wv1dEOV6r75kIN4-cxXtFDbgHL9ciQQ_x4,20131
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=AHU88W-UJ5alHOA3pN2oq7lIMiyYWdFSM0BL2WR-FGs,20998
 code_loader/leaploader.py,sha256=Tpf6A25hYuo4D0umGL3BHNYJhmz_NIwvFveQgAlsSOo,19534
 code_loader/utils.py,sha256=aw2i_fqW_ADjLB66FWZd9DfpCQ7mPdMyauROC5Nd51I,2197
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=VoqO9FN84yXyMjRjHjUTOt2GdTkJRMbHbXJ1cJkREkk,2230
-code_loader-1.0.60.dev2.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.60.dev2.dist-info/METADATA,sha256=Qd_qx-0HBEh0g-Yimq2xX-aQ0CJwpNuPSUDNVCmKcwA,893
-code_loader-1.0.60.dev2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.60.dev2.dist-info/RECORD,,
+code_loader-1.0.62.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.62.dist-info/METADATA,sha256=s-UxETRAemR6kd81be7T9OdR7JCIjI4uIXisVRBZORc,888
+code_loader-1.0.62.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.62.dist-info/RECORD,,