From 900ebfcd510162c50d35dc2be11feebbc78957d3 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 19 Sep 2019 15:53:07 +0200
Subject: [PATCH] Test all variants on CI

---
 tests/test_tfmad.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_tfmad.py b/tests/test_tfmad.py
index 46279dd..5534819 100644
--- a/tests/test_tfmad.py
+++ b/tests/test_tfmad.py
@@ -239,8 +239,8 @@ def test_tfmad_gradient_check_torch_native(with_offsets, with_cuda):
         [dict[f] for f in auto_diff.forward_input_fields]), atol=1e-4, raise_exception=True)
 
 
-# @pytest.mark.parametrize('with_offsets, with_cuda', itertools.product((False, True), repeat=2))
-@pytest.mark.parametrize('with_offsets, with_cuda, gradient_check', ((False, False, True),))
+# @pytest.mark.parametrize('with_offsets, with_cuda, gradient_check', ((True, False, False),))
+@pytest.mark.parametrize('with_offsets, with_cuda', itertools.product((False, True), repeat=3))
 def test_tfmad_gradient_check_tensorflow_native(with_offsets, with_cuda, gradient_check):
     pytest.importorskip('tensorflow')
     import tensorflow as tf
-- 
GitLab