Update tests/ignite/engine/test_create_supervised.py

pytorch · Dec 9, 2024 · 7bee93e · 7bee93e
1 parent 1b0ecc2
commit 7bee93e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/ignite/engine/test_create_supervised.py b/tests/ignite/engine/test_create_supervised.py
@@ -447,7 +447,7 @@ def test_create_supervised_trainer_apex_error():
 def mock_torch_cuda_amp_module():
     with patch.dict(
         "sys.modules",
-        {"torch.amp": None, "torch.cuda.amp.grad_scaler": None, "torch.amp.autocast_mode": None},
+        {"torch.amp": None, "torch.cuda.amp": None, "torch.amp.autocast_mode": None},
     ):
         yield torch