fix tests

mosaicml · Oct 25, 2024 · ded11da · ded11da
1 parent eb05ed5
commit ded11da
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tests/utils/test_autolog_hparams.py b/tests/utils/test_autolog_hparams.py
@@ -176,8 +176,6 @@ def test_extract_hparams_trainer():
 
         # DeepSpeed
         'deepspeed_config': None,
-        'fsdp_config': None,
-        'fsdp_auto_wrap': True,
         'parallelism_config': None,
 
         # System/Numerics