Try to fix tests

Netflix · Nov 19, 2024 · f7d8543 · f7d8543
1 parent 407c701
commit f7d8543
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/metaflow/cmd/develop/stub_generator.py b/metaflow/cmd/develop/stub_generator.py
@@ -1243,7 +1243,10 @@ def exploit_default(default_value: Any) -> Optional[str]:
 
                 default = exploit_default(parameter.default)
 
-                if kw_only_param and parameter.kind != inspect.Parameter.KEYWORD_ONLY:
+                if kw_only_param and parameter.kind not in (
+                    inspect.Parameter.KEYWORD_ONLY,
+                    inspect.Parameter.VAR_KEYWORD,
+                ):
                     raise RuntimeError(
                         "In function '%s': cannot have a positional parameter after a "
                         "keyword only parameter" % name

diff --git a/test/core/run_tests.py b/test/core/run_tests.py
@@ -11,7 +11,8 @@
 from multiprocessing import Pool
 
 from metaflow._vendor import click
-from metaflow.cli import run, start
+from metaflow.cli import start
+from metaflow.cli_components.run_cmds import run
 
 skip_api_executor = False