fixing CI errors

davidliujiafeng · Jul 19, 2022 · 9ac99e5 · 9ac99e5
1 parent 2c81915
commit 9ac99e5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -21,6 +21,7 @@ jobs:
 
     - name: Install dependencies
       run: |
+        sudo apt-get install -y libsndfile1
         python3 -m venv env
         . env/bin/activate
         python -m pip install --upgrade pip

diff --git a/demucs/repo.py b/demucs/repo.py
@@ -68,7 +68,8 @@ def get_model(self, sig: str) -> Model:
             url = self._models[sig]
         except KeyError:
             raise ModelLoadingError(f'Could not find a pre-trained model with signature {sig}.')
-        pkg = torch.hub.load_state_dict_from_url(url, map_location='cpu', check_hash=True)
+        pkg = torch.hub.load_state_dict_from_url(
+            url, map_location='cpu', check_hash=True)  # type: ignore
         return load_model(pkg)