[test] Fixes EarlyStopping flaky test (#2926)

deepjavalibrary · Jan 9, 2024 · 1cbddbf · 1cbddbf
1 parent d58b755
commit 1cbddbf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...n/src/main/java/ai/djl/integration/tests/training/listener/EarlyStoppingListenerTest.java b/...n/src/main/java/ai/djl/integration/tests/training/listener/EarlyStoppingListenerTest.java
@@ -178,7 +178,7 @@ public void testEarlyStoppingStopsOnEpoch1AsMaxDurationIs1ms() throws Exception
                     // epoch
                     EasyTrain.fit(trainer, 10, trainMnistDataset, testMnistDataset);
                 } catch (EarlyStoppingListener.EarlyStoppedException e) {
-                    Assert.assertTrue(e.getMessage().contains("ms elapsed >= 1 maxMillis"));
+                    Assert.assertTrue(e.getMessage().contains("ms elapsed >="));
                     Assert.assertTrue(e.getStopEpoch() < 10); // Stop epoch is before 10
                 }