Skip to content

Commit ffdbd92

Browse files
committed
fix up tests for numpy 2.0
1 parent adfe9ce commit ffdbd92

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_decode.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ def test_beam_search_mismatched_alphabet_long(self):
9999

100100
def test_nans(self):
101101
"""beam_search is passed NaN values"""
102-
self.probs.fill(np.NaN)
102+
self.probs.fill(np.nan)
103103
with self.assertRaisesRegex(RuntimeError, "Failed to compare values"):
104104
beam_search(self.probs, self.alphabet)
105105

@@ -369,7 +369,7 @@ def get_random_data(self, samples=100):
369369

370370
def test_nans(self):
371371
"""beam_search_duplex is passed NaN values"""
372-
self.probs_1.fill(np.NaN)
372+
self.probs_1.fill(np.nan)
373373
with self.assertRaisesRegex(RuntimeError, "Failed to compare values"):
374374
beam_search_duplex(self.probs_1, self.probs_2, self.alphabet)
375375

0 commit comments

Comments
 (0)