450

Lightning-AI · Feb 24, 2021 · c4c4062 · c4c4062
1 parent a0670bb
commit c4c4062
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -92,7 +92,7 @@ jobs:
       - image: circleci/python:3.7
     environment:
       - XLA_VER: 1.7
-      - MAX_CHECKS: 250
+      - MAX_CHECKS: 450
       - CHECK_SPEEP: 10
     steps:
       - checkout

diff --git a/.github/workflows/ci_test-tpu.yml b/.github/workflows/ci_test-tpu.yml
@@ -12,7 +12,7 @@ env:
   GKE_CLUSTER: lightning-cluster
   GKE_ZONE: us-central1-a
   IMAGE: gcr.io/${{ secrets.GKE_PROJECT }}/tpu-testing-image
-  MAX_CHECKS: 300
+  MAX_CHECKS: 450
   CHECK_SPEEP: 10
 
 jobs:

diff --git a/tests/metrics/regression/test_ssim.py b/tests/metrics/regression/test_ssim.py
@@ -49,7 +49,8 @@ def _sk_metric(preds, target, data_range, multichannel):
     )
 
 
-@pytest.mark.parametrize("preds, target, multichannel",
+@pytest.mark.parametrize(
+    "preds, target, multichannel",
     [(i.preds, i.target, i.multichannel) for i in _inputs],
 )
 @pytest.mark.skipif(_TPU_AVAILABLE, reason='Seems to be hanging on TPU, need to be fixed!')  # todo