Skip to content

Commit 84d7d9c

Browse files
committed
fix: enforce non-NaN rows in mock data and restore return values
1 parent aeef5b8 commit 84d7d9c

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

tests/create_sequence_data.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,10 @@ def _generate_sequence_data(
5050
)
5151
data.flat[nan_indices] = np.nan
5252

53+
# ensure each row has at least one non-NaN
54+
row_all_nan = np.isnan(data).all(axis=1)
55+
data[row_all_nan, 0] = 0.0
56+
5357
if not use_mem_mapped:
5458
np.save(sequence_root / "data.npy", data)
5559
else:

0 commit comments

Comments
 (0)