code-loader 1.0.72.dev2__tar.gz → 1.0.72.dev3__tar.gz

{code_loader-1.0.72.dev2 → code_loader-1.0.72.dev3}/PKG-INFO

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

{code_loader-1.0.72.dev2 → code_loader-1.0.72.dev3}/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-1.0.72.dev2 → code_loader-1.0.72.dev3}/pyproject.toml

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