Use numpy.nan instead of the alias np.NAN

AliceO2Group · Jul 11, 2024 · 2aff0e7 · 2aff0e7
1 parent 5eb4579
commit 2aff0e7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/o2tuner/inspector.py b/src/o2tuner/inspector.py
@@ -269,7 +269,7 @@ def plot_correlations(self, *, n_most_important=None):
         param_values = []
         params_plot = []
         for param_key, label in zip(params[::-1], params_labels[::-1]):
-            values = [trial.params.get(param_key, np.NAN) for trial in self._trials_complete]
+            values = [trial.params.get(param_key, np.nan) for trial in self._trials_complete]
             param_values.append(values)
             params_plot.append(label)
 
@@ -304,7 +304,7 @@ def plot_pairwise_scatter(self, *, n_most_important=None):
         param_values = []
         params_plot = []
         for param_key, label in zip(params[::-1], params_labels[::-1]):
-            values = [trial.params.get(param_key, np.NAN) for trial in self._trials_complete]
+            values = [trial.params.get(param_key, np.nan) for trial in self._trials_complete]
             param_values.append(values)
             params_plot.append(label)
 
@@ -336,7 +336,7 @@ def plot_loss_feature_history(self, *, n_most_important=None):
         param_values = []
         params_plot = []
         for param_key, label in zip(params[::-1], params_labels[::-1]):
-            values = [trial.params.get(param_key, np.NAN) for trial in self._trials_complete]
+            values = [trial.params.get(param_key, np.nan) for trial in self._trials_complete]
             param_values.append(values)
             params_plot.append(label)