fix fad calculation for newer versions of scipy (#25)

gudgud96 · Jan 25, 2024 · 99ab594 · 99ab594
1 parent cf50298
commit 99ab594
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/frechet_audio_distance/fad.py b/frechet_audio_distance/fad.py
@@ -327,13 +327,13 @@ def calculate_frechet_distance(self, mu1, sigma1, mu2, sigma2, eps=1e-6):
         diff = mu1 - mu2
 
         # Product might be almost singular
-        covmean, _ = linalg.sqrtm(sigma1.dot(sigma2), disp=False)
+        covmean, _ = linalg.sqrtm(sigma1.dot(sigma2).astype(complex), disp=False)
         if not np.isfinite(covmean).all():
             msg = ('fid calculation produces singular product; '
                    'adding %s to diagonal of cov estimates') % eps
             print(msg)
             offset = np.eye(sigma1.shape[0]) * eps
-            covmean = linalg.sqrtm((sigma1 + offset).dot(sigma2 + offset))
+            covmean = linalg.sqrtm((sigma1 + offset).dot(sigma2 + offset).astype(complex))
 
         # Numerical error might give slight imaginary component
         if np.iscomplexobj(covmean):

diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,6 @@
-numpy==1.23.4
+numpy
 torch
-scipy==1.10.1
+scipy
 tqdm
 soundfile
 resampy