Formatting correctly

asteroid-team · Nov 20, 2020 · d79cfa9 · d79cfa9
1 parent 9ae875e
commit d79cfa9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/test_formatting.yml b/.github/workflows/test_formatting.yml
@@ -16,7 +16,7 @@ jobs:
       - name: Install Black and flake8
         run: pip install black==20.8b1 flake8
       - name: Run Black
-        run: python -m black --config=pyproject.toml --check filterbanks tests
+        run: python -m black --config=pyproject.toml --check asteroid_filterbanks tests
 
       - name: Link with flake8
         # Exit on important linting errors and warn about others.

diff --git a/tests/jit_filterbanks_test.py b/tests/jit_filterbanks_test.py
@@ -47,8 +47,8 @@ def __init__(
         )
         self.encoder = encoder
         self.decoder = decoder
-        
+
     def forward(self, wav):
         tf_rep = self.encoder(wav)
-        wav_back = self.decoder(wav)
-        return wav_back
+        wav_back = self.decoder(tf_rep)
+        return wav_back