Address PR feedback.

ludwig-ai · Jun 10, 2022 · 4cf64e2 · 4cf64e2
1 parent 22813df
commit 4cf64e2
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/ludwig/utils/test_defaults.py b/tests/ludwig/utils/test_defaults.py
@@ -1,5 +1,4 @@
 import copy
-from dataclasses import asdict
 
 import pytest
 
@@ -93,7 +92,7 @@ def test_merge_with_defaults_early_stop(use_train, use_hyperopt_scheduler):
 
     merged_config = merge_with_defaults(config)
 
-    expected = -1 if use_hyperopt_scheduler else asdict(TrainerConfig())["early_stop"]
+    expected = -1 if use_hyperopt_scheduler else TrainerConfig().early_stop
     assert merged_config[TRAINER]["early_stop"] == expected