diff --git a/tests/test_load_dump_pipeline.py b/tests/test_load_dump_pipeline.py index b851218c..4a02f74f 100644 --- a/tests/test_load_dump_pipeline.py +++ b/tests/test_load_dump_pipeline.py @@ -62,7 +62,7 @@ def test_dummytf(self): self.assertTrue(d31.component[0].is_trained) self.assertTrue(d31.component[1].is_trained) - d3.work_dir = './' + d3.work_dir = self.dirname d3.name = 'dummy-pipeline' d3.is_trained = True d3.dump_yaml()