flamo 0.2.1__tar.gz → 0.2.3__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flamo
-Version: 0.2.1
+Version: 0.2.3
 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.3}/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)
@@ -308,6 +309,7 @@ class LossProfile:
         audio distances are bad at pitch,” arXiv preprint arXiv:2012.04572, 2020.
         """
         steps = self.steps
+        steps = steps.cpu().numpy()
         # find the index in steps of the element closest to the target value
         target_indx = np.abs(steps - self.param_config.target_value).argmin()
         accuracy = np.empty(loss.shape)
@@ -470,7 +472,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:
@@ -634,6 +636,8 @@ class LossSurface(LossProfile):
         """
         steps_0 = self.steps_0
         steps_1 = self.steps_1
+        steps_0 = steps_0.cpu().numpy()
+        steps_1 = steps_1.cpu().numpy()
         # find the index in steps of the element closest to the target value
         target_indx_0 = np.abs(steps_0 - self.param_config[0].target_value).argmin()
         target_indx_1 = np.abs(steps_1 - self.param_config[1].target_value).argmin()

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

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