Skip to content

Commit

Permalink
fixup! Fix static checks
Browse files Browse the repository at this point in the history
  • Loading branch information
drasmuss committed Oct 31, 2022
1 parent 8e9ffa6 commit 5bf9804
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion keras_lmu/tests/test_benchmarks.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 5bf9804

Please sign in to comment.