Merge pull request #134 from KinWaiCheuk/update

updated to nnAudio 0.3.3
KinWaiCheuk · Feb 13, 2024 · 9e9a4ba · 9e9a4ba
2 parents ef941af + e9e723f
commit 9e9a4ba
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/.github/workflows/nnAudio.yaml b/.github/workflows/nnAudio.yaml
@@ -4,13 +4,21 @@ on:
     branches:
       - main
       - master
+      - update
+      - workflow
 jobs:
   build:
     runs-on: ubuntu-latest
     strategy:
       matrix:
         python-version: ['3.7', '3.8', '3.9', '3.10']
-        torch-version: ['1.11.0', '1.12.0', '1.13.0']
+        torch-version: ['1.11.0', '1.12.0', '1.13.0', '2.1.0']
+        exclude:
+            # excludes torch 2.1 on older python versions
+            - python-version: '3.7'
+              torch-version: '2.1.0'
+            - python-version: '3.8'
+              torch-version: '2.1.0'              
     steps:
       - uses: actions/checkout@v2
       - name: Set up Python ${{ matrix.python-version }}

diff --git a/Installation/nnAudio/__init__.py b/Installation/nnAudio/__init__.py
@@ -1 +1 @@
-__version__ = "0.3.2"
+__version__ = "0.3.3"
diff --git a/Installation/nnAudio/features/cfp.py b/Installation/nnAudio/features/cfp.py
@@ -143,6 +143,7 @@ def forward(self, x):
             window=self.h,
             onesided=False,
             pad_mode="constant",
+            return_complex=False
         )
         tfr0 = torch.sqrt(tfr0.pow(2).sum(-1)) / torch.norm(
             self.h
@@ -380,6 +381,7 @@ def forward(self, x):
             window=self.h,
             onesided=False,
             pad_mode="constant",
+            return_complex=False
         )
         tfr0 = torch.sqrt(tfr0.pow(2).sum(-1)) / torch.norm(
             self.h