Skip to content

Commit

Permalink
Replace jax.numpy in1d (no longer exists) with isin
Browse files Browse the repository at this point in the history
  • Loading branch information
kysolvik committed Jun 7, 2024
1 parent 3ac5714 commit f3b5015
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dabench/data/enso_indices.py
Original file line number Diff line number Diff line change
Expand Up @@ -275,7 +275,7 @@ def _combine_vals_years(self, all_vals, all_years):
all_vals[v] = all_vals[v][np.sort(indices)]
all_years[v] = all_years[v][np.sort(indices)]
# Append common_vals
common_vals.append(all_vals[v][jnp.in1d(all_years[v],
common_vals.append(all_vals[v][jnp.isin(all_years[v],
common_years)])
for f in common_vals:
print(f.shape)
Expand Down

0 comments on commit f3b5015

Please sign in to comment.