Skip to content
Snippets Groups Projects
Unverified Commit 0d29ec03 authored by Alain Riou's avatar Alain Riou Committed by GitHub
Browse files

Merge pull request #24 from SonyCSLParis/omegaconf

remove omegaconf dependency in checkpoint
parents e8cdebb0 5209902b
Branches
No related tags found
No related merge requests found
......@@ -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
......
......@@ -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
......
......@@ -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
......
No preview for this file type
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment