Merge branch 'main' into keyboard_interrupt

requirements.txt

@@ -34,6 +34,6 @@ sphinxcontrib-qthelp==1.0.3
 sphinxcontrib-serializinghtml==1.1.5
 sympy==1.12
 tomli==2.0.1
-torch==2.0.1
+torch==2.2.0
 tqdm==4.66.3
 typing_extensions==4.7.1

setup.py

@@ -11,7 +11,7 @@
 dependencies = [
     "numpy<2",
     "sympy",
-    "torch>=2.0.1",
+    "torch>=2.2.0",
     "tqdm",
 ]
 

src/simulated_bifurcation/optimizer/simulated_bifurcation_optimizer.py

@@ -5,7 +5,7 @@
 
 import torch
 from numpy import minimum
-from tqdm import tqdm
+from tqdm.auto import tqdm
 
 from .environment import ENVIRONMENT
 from .simulated_bifurcation_engine import SimulatedBifurcationEngine

src/simulated_bifurcation/optimizer/stop_window.py

@@ -1,7 +1,7 @@
 from typing import Tuple, Union
 
 import torch
-from tqdm import tqdm
+from tqdm.auto import tqdm
 
 
 class StopWindow:
@@ -66,7 +66,9 @@ def __init_tensor(self, dtype: torch.dtype) -> torch.Tensor:
         return torch.zeros(self.n_agents, device=self.device, dtype=dtype)
 
     def __init_energies(self) -> None:
-        self.energies = torch.tensor([float("inf") for _ in range(self.n_agents)])
+        self.energies = torch.tensor(
+            [float("inf") for _ in range(self.n_agents)], device=self.device
+        )
 
     def __init_tensors(self) -> None:
         self.stability = self.__init_tensor(torch.int16)