From 5bf98040796975fd7a0cb363d16406ab760ee0db Mon Sep 17 00:00:00 2001 From: Daniel Rasmussen Date: Mon, 31 Oct 2022 15:07:21 -0300 Subject: [PATCH] fixup! Fix static checks --- keras_lmu/tests/test_benchmarks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/keras_lmu/tests/test_benchmarks.py b/keras_lmu/tests/test_benchmarks.py index d58de82e..dffb1832 100644 --- a/keras_lmu/tests/test_benchmarks.py +++ b/keras_lmu/tests/test_benchmarks.py @@ -38,7 +38,7 @@ def on_predict_batch_end(self, batch, logs=None): @pytest.mark.skipif(not tf_gpu_installed, reason="Very slow on CPU") @pytest.mark.parametrize( "mode, min_time, max_time", - [("rnn", 0.1, 0.2), ("fft", 0.1, 0.2), ("raw", 0.05, 0.15)], + [("rnn", 0.1, 0.2), ("fft", 0.05, 0.15), ("raw", 0.05, 0.15)], ) def test_performance(mode, min_time, max_time): # performance is based on Azure NC6 VM