np.infty -> np.inf

ekorman · Jul 25, 2024 · 1ea5661 · 1ea5661
1 parent 0104fca
commit 1ea5661
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/neurve/core/trainer.py b/neurve/core/trainer.py
@@ -91,8 +91,8 @@ def log_dict(self, d):
     def train(
         self,
         n_epochs=None,
-        save_ckpt_freq=np.infty,
-        eval_freq=np.infty,
+        save_ckpt_freq=np.inf,
+        eval_freq=np.inf,
         n_global_steps=None,
     ):
         """Method for training. Saves tensorboard output to self.out_path and

diff --git a/neurve/metric_learning/trainer.py b/neurve/metric_learning/trainer.py
@@ -45,7 +45,7 @@ def log_metrics(self, dists, all_y):
         all_y : np.ndarray
         """
         all_y = np.ones((dists.shape[0], 1)) @ all_y.reshape(1, -1)
-        np.fill_diagonal(dists, np.infty)
+        np.fill_diagonal(dists, np.inf)
         sorted_indices = np.argsort(dists)
 
         # sort dists and labels by distance