fix test

VIDA-NYU · Nov 21, 2023 · d4f2d42 · d4f2d42
1 parent a50ec0b
commit d4f2d42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -152,7 +152,7 @@ class TestExportPipelineAsCode:
 
     def test_write_pipeline_code_as_pyfile(self):
         assert exists("pipeline_99_code.py") == False
-        write_pipeline_code_as_pyfile(self.pipeline_id, self.pipeline_obj)
+        write_pipeline_code_as_pyfile(self.pipeline_id, self.pipeline_obj, "REGRESSION")
 
         assert exists("pipeline_99_code.py") == True
         remove("pipeline_99_code.py")