Revert "Update test_metric_result_integration.py"

This reverts commit bf70e43.
neptune-ai · Mar 24, 2021 · 6c095b2 · 6c095b2
1 parent 250d0aa
commit 6c095b2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/core/test_metric_result_integration.py b/tests/core/test_metric_result_integration.py
@@ -16,6 +16,8 @@
 import torch.distributed as dist
 import torch.multiprocessing as mp
 from torchmetrics import Metric
+import numpy
+import os
 
 import tests.helpers.utils as tutils
 from pytorch_lightning.core.step_result import Result
@@ -96,6 +98,7 @@ def test_result_reduce_ddp():
     tutils.set_random_master_port()
 
     worldsize = 2
+    os.environ["MKL_SERVICE_FORCE_INTEL"] = "1"
     mp.spawn(_ddp_test_fn, args=(worldsize, ), nprocs=worldsize)