code formatting

pytorch · Nov 21, 2023 · 6545133 · 6545133
1 parent d1dfbec
commit 6545133
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tests/ignite/distributed/utils/test_serial.py b/tests/ignite/distributed/utils/test_serial.py
@@ -1,3 +1,4 @@
+import pytest
 import torch
 
 import ignite.distributed as idist

diff --git a/tests/ignite/engine/test_create_supervised.py b/tests/ignite/engine/test_create_supervised.py
@@ -309,7 +309,7 @@ def _test_create_supervised_evaluator(
         if Version(torch.__version__) >= Version("1.7.0"):
             # This is broken in 1.6.0 but will be probably fixed with 1.7.0
             err_msg_1 = "Expected all tensors to be on the same device"
-            err_msg_2 = "Placeholder storage has not been allocated on MPS device" 
+            err_msg_2 = "Placeholder storage has not been allocated on MPS device"
             with pytest.raises(RuntimeError, match=f"({err_msg_1}|{err_msg_2})"):
                 evaluator.run(data)