Skip to content

Commit

Permalink
Update workflows for PyTorch 2.x
Browse files Browse the repository at this point in the history
Signed-off-by: Beat Buesser <[email protected]>
  • Loading branch information
beat-buesser committed Dec 7, 2023
1 parent 586cd39 commit 6787d0a
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 3 deletions.
3 changes: 3 additions & 0 deletions .github/actions/deepspeech-v3/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ RUN apt-get update \
libsndfile-dev \
libsndfile1 \
vim \
curl \
&& apt-get clean all \
&& rm -r /var/lib/apt/lists/*

Expand All @@ -30,3 +31,5 @@ RUN cd deepspeech.pytorch && sed -i '/^sklearn/d' requirements.txt && pip instal

RUN pip install numba==0.56.4 pytest-cov==4.1.0 pydub==0.25.1
RUN pip list

RUN mkdir -p /root/.art/data && cd /root/.art/data && curl -LJO "https://github.com/SeanNaren/deepspeech.pytorch/releases/download/V3.0/librispeech_pretrained_v3.ckpt"
4 changes: 2 additions & 2 deletions .github/workflows/ci-style-checks.yml
Original file line number Diff line number Diff line change
Expand Up @@ -31,15 +31,15 @@ jobs:
- name: Setup Python
uses: actions/setup-python@v4
with:
python-version: 3.8
python-version: '3.10'
- name: Pre-install
run: |
sudo apt-get update
sudo apt-get -y -q install ffmpeg libavcodec-extra
- name: Install Dependencies
run: |
python -m pip install --upgrade pip setuptools wheel
pip install -q pylint==2.12.2 mypy==0.931 pycodestyle==2.8.0 black==21.12b0
pip install -q pylint==2.12.2 mypy==1.7.1 pycodestyle==2.8.0 black==21.12b0
pip install -q -r <(sed '/^numpy/d;/^pluggy/d;/^tensorflow/d;/^keras/d' requirements_test.txt)
pip install numpy==1.22.4
pip install pluggy==0.13.1
Expand Down
2 changes: 1 addition & 1 deletion art/defences/transformer/poisoning/strip.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ def __call__( # type: ignore
self,
num_samples: int = 20,
false_acceptance_rate: float = 0.01,
) -> CLASSIFIER_TYPE:
) -> "CLASSIFIER_TYPE":
"""
Create a STRIP defense
Expand Down

0 comments on commit 6787d0a

Please sign in to comment.