diff --git a/.github/workflows/nnAudio.yaml b/.github/workflows/nnAudio.yaml index 15523bf..a80b761 100644 --- 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 index f9aa3e1..e19434e 100755 --- 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 index d6c81ce..9bc3f8e 100644 --- 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