Skip to content

Commit b2b8687

Browse files
colin2328facebook-github-bot
authored andcommitted
bump version and pin requirements for release; add 3.10 to upload (#897)
Summary: Pull Request resolved: #897 as title Reviewed By: osalpekar Differential Revision: D42057947 fbshipit-source-id: c1a965e92443d9a96ad895621cd45f39c8b18449
1 parent d06112f commit b2b8687

File tree

4 files changed

+47
-53
lines changed

4 files changed

+47
-53
lines changed

.github/workflows/nightly_build.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,7 @@ jobs:
111111
strategy:
112112
matrix:
113113
os: [linux.4xlarge.nvidia.gpu]
114-
python-version: [3.7, 3.8, 3.9]
114+
python-version: [3.7, 3.8, 3.9, "3.10"]
115115
cuda-tag: ["cu11"]
116116
needs: build_on_cpu
117117
# the glibc version should match the version of the one we used to build the binary

.github/workflows/release_build.yml

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -101,13 +101,13 @@ jobs:
101101
name: torchrec_${{ matrix.python-version }}_${{ matrix.cuda-tag }}.whl
102102
path: dist/torchrec-*.whl
103103

104-
# download from GHA, test on gpu and push to pypi
105-
test_on_gpu:
104+
# download from GHA, sanity check on gpu and push to pypi
105+
sanity_check_on_gpu_and_push:
106106
runs-on: ${{ matrix.os }}
107107
strategy:
108108
matrix:
109109
os: [linux.4xlarge.nvidia.gpu]
110-
python-version: [3.7, 3.8, 3.9]
110+
python-version: [3.7, 3.8, 3.9, "3.10"]
111111
cuda-tag: ["cu11"]
112112
needs: build_on_cpu
113113
# the glibc version should match the version of the one we used to build the binary
@@ -190,12 +190,6 @@ jobs:
190190
run: |
191191
conda run -n build_binary \
192192
python -c "import torchrec"
193-
- name: Test with pytest
194-
run: |
195-
conda run -n build_binary \
196-
python -m pip install pytest
197-
conda run -n build_binary \
198-
python -m pytest torchrec -v -s -W ignore::pytest.PytestCollectionWarning --continue-on-collection-errors
199193
# Push to Pypi
200194
- name: Push TorchRec Binary to PYPI
201195
env:

requirements.txt

Lines changed: 42 additions & 42 deletions
Original file line numberDiff line numberDiff line change
@@ -1,43 +1,43 @@
1-
arrow
2-
attrs
3-
certifi
4-
charset-normalizer
5-
cmake
6-
Cython
7-
distro
8-
docker
9-
docstring-parser
1+
arrow>=1.2.3
2+
attrs>=22.1.0
3+
certifi>=2022.12.7
4+
charset-normalizer>=2.1.1
5+
cmake>=3.25.0
6+
Cython>=0.29.32
7+
distro>=1.8.0
8+
docker>=6.0.1
9+
docstring-parser>=0.8.1
1010
fbgemm-gpu-nightly
11-
filelock
12-
fsspec
13-
hypothesis
14-
idna
15-
iopath
16-
Jinja2
17-
MarkupSafe
18-
mypy-extensions
19-
ninja
20-
numpy
21-
packaging
22-
pandas
23-
portalocker
24-
pyarrow
25-
pyDeprecate
26-
pyparsing
27-
pyre-extensions==0.0.27
28-
python-dateutil
29-
pytz
30-
PyYAML
31-
requests
32-
scikit-build
33-
six
34-
sortedcontainers
35-
tabulate
36-
torchmetrics
37-
torchx
38-
tqdm
39-
typing-inspect
40-
typing_extensions
41-
urllib3
42-
usort
43-
websocket-client
11+
filelock>=3.8.2
12+
fsspec>=2022.11.0
13+
hypothesis>=6.61.0
14+
idna>=3.4
15+
iopath>=0.1.10
16+
Jinja2>=3.1.2
17+
MarkupSafe>=2.1.1
18+
mypy-extensions>=0.4.3
19+
ninja>=1.11.1
20+
numpy>=1.21.6
21+
packaging>=22.0
22+
pandas>=1.3.5
23+
portalocker>=2.6.0
24+
pyarrow>=10.0.1
25+
pyDeprecate>=0.3.2
26+
pyparsing>=3.0.9
27+
pyre-extensions>=0.0.27
28+
python-dateutil>=2.8.2
29+
pytz>=2022.6
30+
PyYAML>=6.0
31+
requests>=2.28.1
32+
scikit-build>=0.16.3
33+
six>=1.16.0
34+
sortedcontainers>=2.4.0
35+
tabulate>=0.9.0
36+
torchmetrics>=0.11.0
37+
torchx>=0.3.0
38+
tqdm>=4.64.1
39+
typing-inspect>=0.8.0
40+
typing_extensions>=4.4.0
41+
urllib3>=1.26.13
42+
usort>=1.0.5
43+
websocket-client>=1.4.2

version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,4 +14,4 @@
1414
# 0.1.0bN # Beta release
1515
# 0.1.0rcN # Release Candidate
1616
# 0.1.0 # Final release
17-
__version__ = "0.3.01"
17+
__version__ = "0.3.2"

0 commit comments

Comments
 (0)