code-loader 1.0.72.dev1__py3-none-any.whl → 1.0.72.dev3__py3-none-any.whl

code_loader/inner_leap_binder/leapbinder.py

@@ -240,15 +240,14 @@ class LeapBinder:
             leap_binder.add_custom_loss(custom_loss_function, name='custom_loss')
         """
 
-        regular_arg_names = []
+        regular_arg_names = inspect.getfullargspec(function)[0]
         preprocess_response_arg_name = None
         for arg_name, arg_type in inspect.getfullargspec(function).annotations.items():
             if arg_type == SamplePreprocessResponse:
                 if preprocess_response_arg_name is not None:
                     raise Exception("only one argument can be of type SamplePreprocessResponse")
                 preprocess_response_arg_name = arg_name
-            else:
-                regular_arg_names.append(arg_name)
+                regular_arg_names.remove(arg_name)
 
         self.setup_container.custom_loss_handlers.append(
             CustomLossHandler(CustomLossHandlerData(name, regular_arg_names), function))

code_loader/leaploader.py

@@ -330,8 +330,7 @@ class LeapLoader(LeapLoaderBase):
             for visualizer_handler in setup.visualizers]
 
         custom_losses = [CustomLossInstance(custom_loss.custom_loss_handler_data.name,
-                                            custom_loss.custom_loss_handler_data.arg_names,
-                                            custom_loss.custom_loss_handler_data.preprocess_response_arg_name)
+                                            custom_loss.custom_loss_handler_data.arg_names)
                          for custom_loss in setup.custom_loss_handlers]
 
         prediction_types = []

{code_loader-1.0.72.dev1.dist-info → code_loader-1.0.72.dev3.dist-info}/METADATA

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

{code_loader-1.0.72.dev1.dist-info → code_loader-1.0.72.dev3.dist-info}/RECORD

@@ -19,14 +19,14 @@ code_loader/experiment_api/types.py,sha256=MY8xFARHwdVA7p4dxyhD60ShmttgTvb4qdp1o
 code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZJEqBqc,3304
 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=una-6k_nnSeqrLNjCy3beXdje_MWXbCRLoCalKdNYbg,26693
+code_loader/inner_leap_binder/leapbinder.py,sha256=-fryKzD8T8K2EgrOsR5NryabP8_1k_m3POLwhYIA_8I,26708
 code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=ebMxknpKMW-dE8Erq0fFq4RrE5E_Jfx9IvmRRZSdhlc,20813
-code_loader/leaploader.py,sha256=DCrSS2789dkkePROJSwhAhgzQvR4_27YAR__PxCcLNc,23025
+code_loader/leaploader.py,sha256=g160Z0MRop_7m3bu1HawcGoZNKakvs9Su4tEfbQ9pR0,22914
 code_loader/leaploaderbase.py,sha256=FDXjTFBjnpfUyZ2tS3wLo7y2j82Qb_VKHI1BMB5gFww,3269
 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=Ffx5VHVOe5ujBOsjBSxN_aIEVwFSQ6gbhTMG5aUS-po,2305
-code_loader-1.0.72.dev1.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.72.dev1.dist-info/METADATA,sha256=WBFziucTTs-QHppp3QzdoSp_j9KXh1W9njNsHAGxxSo,854
-code_loader-1.0.72.dev1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.72.dev1.dist-info/RECORD,,
+code_loader-1.0.72.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.72.dev3.dist-info/METADATA,sha256=-9q2S44uQTVZhb3uYyUp_T3ByP5WuljvF9blRuuly3g,854
+code_loader-1.0.72.dev3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.72.dev3.dist-info/RECORD,,