Apply isort and black reformatting

Signed-off-by: ShriyaPalsamudram <[email protected]>
NVIDIA · Nov 12, 2024 · a8fda44 · a8fda44
1 parent 3f92c72
commit a8fda44
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/collections/llm/test_fault_tolerance.py b/tests/collections/llm/test_fault_tolerance.py
@@ -42,10 +42,10 @@ def on_train_batch_end(self, trainer, pl_module, outputs, batch, batch_idx):
         if self.crash_step and trainer.global_step == self.crash_step:
             raise Exception(f"Simulating a crash at step {self.crash_step}!")
 
-        #if (
+        # if (
         #    datetime.now() <= datetime.strptime(self.crash_time, "%Y-%m-%d %H:%M:%S")
         #    and trainer.global_step >= self.crash_step
-        #):
+        # ):
         #    raise Exception("Simulating a crash!")