minor test fix

mlrun · Jun 10, 2024 · 2ca8da5 · 2ca8da5
1 parent 18d62a7
commit 2ca8da5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/describe/test_describe.py b/describe/test_describe.py
@@ -318,7 +318,7 @@ def test_import_function_describe_dask():
 def test_code_to_function_describe_dask():
     dask_uri = "dask_func.yaml"
     _create_dask_func(dask_uri)
-    describe_func = code_to_function(filename="describe.py", kind="local")
+    describe_func = import_function("function.yaml")
     is_test_passed = True
     _create_data(n_samples=100, n_features=5, n_classes=3, n_informative=3)
     describe_func.spec.command = "describe_dask.py"