From a8fda443729da76d73c939556448ad236c7fbef9 Mon Sep 17 00:00:00 2001 From: ShriyaPalsamudram Date: Tue, 12 Nov 2024 17:42:31 +0000 Subject: [PATCH] Apply isort and black reformatting Signed-off-by: ShriyaPalsamudram --- tests/collections/llm/test_fault_tolerance.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/collections/llm/test_fault_tolerance.py b/tests/collections/llm/test_fault_tolerance.py index a869f30425264..d70828ebd8fd6 100644 --- 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!")