Merge pull request #24 from SonyCSLParis/omegaconf

remove omegaconf dependency in checkpoint

.github/workflows/test-workflow.yml

@@ -2,7 +2,7 @@ name: Test Workflow
 
 on:
   pull_request:
-    branches: [ "master" ]
+    branches: [ "dev" ]
 
 jobs:
   build:
@@ -21,7 +21,10 @@ jobs:
         python-version: ${{ matrix.python-version }}
     - name: Install dependencies
       run: |
+        sudo apt-get install -y libsox-dev
         python -m pip install --upgrade pip
+        python -m pip install torch torchaudio --index-url https://download.pytorch.org/whl/cpu
+        python -m pip install matplotlib
         python -m pip install pytest
         python -m pip install .
     - name: Test with pytest

pesto/model.py

@@ -161,7 +161,7 @@ class PESTO(nn.Module):
     def __init__(self,
                  encoder: nn.Module,
                  preprocessor: nn.Module,
-                 crop_kwargs: Mapping[str, Any] | None = None,
+                 crop_kwargs: Optional[Mapping[str, Any]] = None,
                  reduction: str = "alwa"):
         super(PESTO, self).__init__()
         self.encoder = encoder

pesto/utils/hcqt.py

@@ -364,7 +364,7 @@ class HarmonicCQT(nn.Module):
             sr: int = 22050,
             hop_length: int = 512,
             fmin: float = 32.7,
-            fmax: float | None = None,
+            fmax: Optional[float] = None,
             bins_per_semitone: int = 1,
             n_bins: int = 84,
             center_bins: bool = True