Ci fix for pytorch profiler log to file tests

Lightning-AI · Apr 13, 2021 · 4c2005d · 4c2005d
1 parent e8b48f1
commit 4c2005d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/tests/test_profiler.py b/tests/test_profiler.py
@@ -174,8 +174,10 @@ def test_advanced_profiler_describe(tmpdir, advanced_profiler):
     # record at least one event
     with advanced_profiler.profile("test"):
         pass
-    # log to stdout and print to file
+    # logs to output file
     advanced_profiler.describe()
+    # ensures file is flushed before reading
+    advanced_profiler.output_file.close()
     data = Path(advanced_profiler.output_fname).read_text()
     assert len(data) > 0
 

diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
@@ -1548,8 +1548,10 @@ def test_pytorch_profiler_describe(pytorch_profiler):
     with pytorch_profiler.profile("test_step"):
         pass
 
-    # log to stdout and print to file
+    # logs to output file
     pytorch_profiler.describe()
+    # ensures file is flushed before reading
+    pytorch_profiler.output_file.close()
     data = Path(pytorch_profiler.output_fname).read_text()
     assert len(data) > 0