fixup! Support for TensorFlow 2.4

nengo · Nov 3, 2020 · c542769 · c542769
1 parent 636b2f8
commit c542769
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/nengo_dl/tests/test_simulator.py b/nengo_dl/tests/test_simulator.py
@@ -832,7 +832,16 @@ def test_tensorboard(Simulator, tmp_path):
 
 @pytest.mark.parametrize("mode", ("predict", "train"))
 @pytest.mark.training
-def test_profile(Simulator, mode, tmp_path):
+def test_profile(Simulator, mode, tmp_path, pytestconfig):
+    if (
+        pytestconfig.getoption("--graph-mode")
+        and version.parse(tf.__version__) >= version.parse("2.4.0rc0")
+        and mode == "predict"
+    ):
+        pytest.skip(
+            "TensorFlow bug, see https://github.com/tensorflow/tensorflow/issues/44563"
+        )
+
     net, a, p = dummies.linear_net()
 
     with Simulator(net) as sim: