[numpy] Fix users of NumPy APIs that are removed in NumPy 2.0.

This change migrates users of APIs removed in NumPy 2.0 to their recommended replacements (https://numpy.org/devdocs/numpy_2_0_migration_guide.html). PiperOrigin-RevId: 655978987
google · Jul 25, 2024 · d5b6d75 · d5b6d75
1 parent c31e6b9
commit d5b6d75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jax_cfd/data/evaluation.py b/jax_cfd/data/evaluation.py
@@ -279,7 +279,7 @@ def compute_summary_dataset(
     else:
       # if the trajectory goes to nan, it just reports a large number
       u_t_corr_sum = [
-          xarray.ones_like(u_x_corr_sum[0]).rename('autocorrelation') * np.infty
+          xarray.ones_like(u_x_corr_sum[0]).rename('autocorrelation') * np.inf
           for threshold in [0.5]
       ]
     energy_sum = [