Revert "Update test_results.py"

This reverts commit ea74906.
shuyingsunshine21 · Mar 24, 2021 · 250d0aa · 250d0aa
1 parent 15f4b9e
commit 250d0aa
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/core/test_results.py b/tests/core/test_results.py
@@ -26,6 +26,8 @@
 from pytorch_lightning.trainer.states import TrainerState
 from tests.helpers import BoringDataModule, BoringModel
 from tests.helpers.runif import RunIf
+import os
+import numpy
 
 
 def _setup_ddp(rank, worldsize):
@@ -50,6 +52,7 @@ def _ddp_test_fn(rank, worldsize, result_cls: Result):
 def test_result_reduce_ddp():
     """Make sure result logging works with DDP"""
     tutils.reset_seed()
+    os.environ["MKL_SERVICE_FORCE_INTEL"] = "1"
     tutils.set_random_master_port()
 
     worldsize = 2