fmt

Signed-off-by: Will Johnson <[email protected]>
foundation-model-stack · Oct 22, 2024 · ace14c7 · ace14c7
1 parent 5d18773
commit ace14c7
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/trackers/test_timing_tracker.py b/tests/trackers/test_timing_tracker.py
@@ -18,8 +18,8 @@
 
 # Standard
 import copy
-import tempfile
 import os
+import tempfile
 
 # First Party
 from tests.test_sft_trainer import (
@@ -29,7 +29,6 @@
     _get_checkpoint_path,
     _test_run_causallm_ft,
     _test_run_inference,
-    _validate_training,
 )
 
 # Local
@@ -74,4 +73,4 @@ def test_sample_run_with_timing_tracker_updated_filename():
             train_log_contents = f.read()
         assert os.path.exists(train_logs_file_path) is True
         assert os.path.getsize(train_logs_file_path) > 0
-        assert "train_runtime" in train_log_contents
+        assert "train_runtime" in train_log_contents