From f3b5015aefc53998cea6a9a95fc31e5d0ba3f882 Mon Sep 17 00:00:00 2001 From: Kylen Solvik Date: Fri, 7 Jun 2024 17:39:10 -0600 Subject: [PATCH] Replace jax.numpy in1d (no longer exists) with isin --- dabench/data/enso_indices.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dabench/data/enso_indices.py b/dabench/data/enso_indices.py index eed23fd..e5b66f1 100644 --- a/dabench/data/enso_indices.py +++ b/dabench/data/enso_indices.py @@ -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)