Skip to content

Commit 1a342a3

Browse files
jbittonzpapakipos
andauthored
Update torchaudio to 0.9 for tensorflow-gpu compatibility (#43)
* Update `torchaudio` to 0.9 for `tensorflow-gpu` compatibility Summary: As called out in #28, there are some conflicting dependencies between `torchaudio`/`torch` 0.8.1/1.8.1 and `tensorflow-gpu`. However, as discovered in pytorch/audio#1595, upgrading to v0.9 etc actually resolve this issue. Thus, I update the torchaudio/torch versions in our `requirements.txt` and I also updated our `numpy` requirement so there are no conflicting dependencies between `tf-gpu` and `augly` :) I verified on my side that all unit tests still pass and that `setup.py` finishes as expected with no errors. I also update `setup.py` to add our README to our PyPI page. Differential Revision: D29292956 fbshipit-source-id: e07f8b3d6d2d8bc9b21af166307f2ae00dbca663 * Update setup.py Co-authored-by: Zoe Papakipos <[email protected]>
1 parent 3de9a97 commit 1a342a3

File tree

2 files changed

+10
-7
lines changed

2 files changed

+10
-7
lines changed

requirements.txt

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,15 +17,14 @@ llvmlite==0.36.0
1717
matplotlib==3.3.4
1818
nlpaug==1.1.3
1919
numba==0.53.1
20-
numpy==1.20.2
20+
numpy>=1.19.5
2121
olefile==0.46
2222
opencv-python==4.5.2.54
2323
packaging==20.9
2424
Pillow==8.2.0
2525
pooch==1.3.0
2626
portalocker==2.3.0
2727
pycparser==2.20
28-
pyOpenSSL==20.0.1
2928
pyparsing==2.4.7
3029
PySocks==1.7.1
3130
python-dateutil==2.8.1
@@ -39,8 +38,8 @@ scipy==1.6.2
3938
six==1.15.0
4039
SoundFile==0.10.3.post1
4140
threadpoolctl==2.1.0
42-
torch==1.8.1
43-
torchaudio==0.8.1
41+
torch==1.9.0
42+
torchaudio==0.9.0
4443
tornado==6.1
4544
tqdm==4.59.0
4645
typing-extensions==3.7.4.3

setup.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,20 +3,24 @@
33

44
import setuptools
55
from pathlib import Path
6-
from typing import Dict
76

87

98
requirements = [
109
r
11-
for r in Path(f"requirements.txt").read_text().splitlines()
10+
for r in Path("requirements.txt").read_text().splitlines()
1211
if '@' not in r
1312
]
1413

14+
with open("README.md", encoding="utf8") as f:
15+
readme = f.read()
16+
1517

1618
setuptools.setup(
1719
name="augly",
18-
version="0.1.1",
20+
version="0.1.2",
1921
description="A data augmentations library for audio, image, text, & video.",
22+
long_description=readme,
23+
long_description_content_type="text/markdown",
2024
url="https://github.com/facebookresearch/AugLy",
2125
author="Zoe Papakipos and Joanna Bitton",
2226
author_email="[email protected]",

0 commit comments

Comments
 (0)