Skip to content

Commit 506c1f5

Browse files
committed
fixup! TMP debug
1 parent 835907b commit 506c1f5

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

.github/workflows/ci.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ jobs:
4040
include:
4141
- script: test
4242
coverage-name: gpu
43-
tf-version: tensorflow[and-cuda]
43+
tf-version: tensorflow[and-cuda]==2.16.1
4444
runs-on: [self-hosted, gpu]
4545
# - script: test
4646
# python-version: "3.11"

keras_lmu/tests/test_layers.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -118,6 +118,7 @@ def hidden_cell():
118118
assert np.allclose(cell_out, layer_out, atol=3e-6 if feedforward else 1e-8), np.max(
119119
np.abs(cell_out - layer_out)
120120
)
121+
assert False
121122

122123

123124
@pytest.mark.parametrize("discretizer", ("zoh", "euler"))

0 commit comments

Comments
 (0)