flamo 0.2.1__tar.gz → 0.2.2__tar.gz

{flamo-0.2.1 → flamo-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flamo
-Version: 0.2.1
+Version: 0.2.2
 Summary: An Open-Source Library for Frequency-Domain Differentiable Audio Processing
 Project-URL: Homepage, https://github.com/gdalsanto/flamo
 Project-URL: Issues, https://github.com/gdalsanto/flamo/issues

{flamo-0.2.1 → flamo-0.2.2}/flamo/optimize/surface.py

@@ -142,6 +142,7 @@ class LossProfile:
                 figsize=((len(self.criteria) + 2) // 3 * 5, 15),
             )
         steps = self.steps
+        steps = steps.cpu().numpy()
         for i_crit in range(len(self.criteria)):
             mean_loss = loss[:, :, i_crit].mean(0)
             std_loss = loss[:, :, i_crit].std(0)
@@ -470,7 +471,7 @@ class LossSurface(LossProfile):
         steps_0 = self.steps_0
         steps_1 = self.steps_1
         # create a meshgrid
-        steps_0, steps_1 = np.meshgrid(steps_0, steps_1)
+        steps_0, steps_1 = np.meshgrid(steps_0.cpu().numpy(), steps_1.cpu().numpy())
         title = ["mean", "std"]
         for i_plot in range(2):
             if len(self.criteria) == 1:

{flamo-0.2.1 → flamo-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "flamo"
-version = "0.2.1"
+version = "0.2.2"
 authors = [
   { name="Gloria Dal Santo", email="gloria.dalsanto@aalto.fi" },
   { name="Gian Marco De Bortoli", email="gian.debortoli@aalto.fi"},