fix failed test

janhenriklambrechts · May 17, 2021 · 50b34c6 · 50b34c6
1 parent c2d6706
commit 50b34c6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_profiler.py b/tests/test_profiler.py
@@ -129,10 +129,11 @@ def test_simple_profiler_log_dir(tmpdir):
     )
     trainer.fit(model)
 
-    expected = tmpdir / "lightning_logs" / "version_0"
+    expected = os.path.join(tmpdir, "lightning_logs/version_0/")
     assert trainer.log_dir == expected
     assert profiler._log_dir == trainer.log_dir
-    assert expected.join("fit-profiler.txt").exists()
+    expected_file_path = os.path.join(tmpdir, "lightning_logs/version_0/fit-profiler.txt")
+    assert os.path.exists(expected_file_path)
 
 
 @RunIf(skip_windows=True)