From 62f0bea8d9f85d4becb52631f29dd0563d2cd72a Mon Sep 17 00:00:00 2001
From: v-chen_data <v-chen_data@example.com>
Date: Sat, 30 Nov 2024 18:00:06 -0500
Subject: [PATCH] revert

---
 tests/test_events.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_events.py b/tests/test_events.py
index 601448e2a2..f15dcb1fcb 100644
--- a/tests/test_events.py
+++ b/tests/test_events.py
@@ -57,7 +57,7 @@ def get_trainer(self, precision='fp32', **kwargs):
             precision=precision,
             train_subset_num_batches=self.train_subset_num_batches,
             eval_subset_num_batches=self.eval_subset_num_batches,
-            max_duration='1ep',
+            max_duration='2ep',
             optimizers=optimizer,
             callbacks=[EventCounterCallback()],
             **kwargs,