Skip to content

Commit 5763687

Browse files
committed
Ported from Kiel: use kwargs for function_args
1 parent ef6e320 commit 5763687

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

dataikuapi/dss/recipe.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1027,7 +1027,7 @@ def __init__(self, name, project):
10271027
ds.write_with_schema(df)
10281028
"""
10291029

1030-
def with_function_name(self, module_name, function_name, function_args=None, custom_template=None):
1030+
def with_function_name(self, module_name, function_name, custom_template=None, **function_args):
10311031
"""
10321032
Defines this recipe as being a functional recipe calling a function name from a module name
10331033
"""
@@ -1041,12 +1041,12 @@ def with_function_name(self, module_name, function_name, function_args=None, cus
10411041

10421042
return self
10431043

1044-
def with_function(self, fn, function_args=None, custom_template=None):
1044+
def with_function(self, fn, custom_template=None, **function_args):
10451045
import inspect
10461046
#TODO: add in documentation that relative imports wont work
10471047
module_name = inspect.getmodule(fn).__name__
10481048
fname = fn.__name__
1049-
return self.with_function_name(module_name, fname, function_args, custom_template)
1049+
return self.with_function_name(module_name, fname, custom_template, **function_args)
10501050

10511051
class SQLQueryRecipeCreator(SingleOutputRecipeCreator):
10521052
"""

0 commit comments

Comments
 (0)