v3.0.2

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "wecopttool"
-version = "3.0.1"
+version = "3.0.2"
 description = "WEC Design Optimization Toolbox"
 readme = "README.md"
 authors = [

tests/test_integration.py

@@ -363,7 +363,7 @@ def test_unstructured_controller_long_crested_wave(self,
         Fex = res_fd[0].force.sel(
             type=['Froude_Krylov', 'diffraction']).sum('type')
         power_optimal = (np.abs(Fex)**2/8 / np.real(hydro_impedance.squeeze())
-                         ).squeeze().sum('omega').item()
+                            ).squeeze().sum('omega').item()
 
         assert power_sol == approx(power_optimal, rel=1e-2)
 

wecopttool/core.py

@@ -758,6 +758,11 @@ def solve(self,
             _log.info("Solving pseudo-spectral control problem "
                       + f"for realization number {realization}.")
 
+            try:
+                wave = wave.squeeze(dim='realization')
+            except KeyError:
+                pass
+
             # objective function
             sign = -1.0 if maximize else 1.0