Skip to content

Commit 12cc5df

Browse files
TroyGardenfacebook-github-bot
authored andcommitted
add psutil in requirements.txt (#3601)
Summary: # context * a recent change in torch/distributed adds dependency on psutil, which is already in PyTorch's requirements.txt {F1984038767} {F1984038775} * add `psutil` in TorchRec's requirements Differential Revision: D88755938
1 parent 3a1d5f3 commit 12cc5df

File tree

5 files changed

+16
-6
lines changed

5 files changed

+16
-6
lines changed

.github/scripts/install_libs.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,4 +28,4 @@ elif [ "$CHANNEL" = "test" ]; then
2828
fi
2929

3030

31-
${CONDA_RUN} pip install importlib-metadata click PyYAML
31+
${CONDA_RUN} pip install -r requirements.txt

.github/workflows/docs.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,3 +122,7 @@ jobs:
122122
if-no-files-found: error
123123
path: docs
124124
s3-prefix: meta-pytorch/torchrec/${{ github.event.pull_request.number }}
125+
126+
concurrency:
127+
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}
128+
cancel-in-progress: true

.github/workflows/unittest_ci.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,3 +136,7 @@ jobs:
136136
--ignore=torchrec/distributed/tests/test_infer_utils.py --ignore=torchrec/distributed/tests/test_fx_jit.py --ignore-glob=**/test_utils/ \
137137
--ignore-glob='*test_train_pipeline*' --ignore=torchrec/distributed/tests/test_model_parallel_hierarchical.py \
138138
-k "$skip_expression"
139+
140+
concurrency:
141+
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}
142+
cancel-in-progress: true

.github/workflows/unittest_ci_cpu.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -115,3 +115,7 @@ jobs:
115115
conda run -n build_binary \
116116
python -m pytest torchrec -v -s -W ignore::pytest.PytestCollectionWarning --continue-on-collection-errors \
117117
--ignore-glob=**/test_utils/ -k "$skip_expression"
118+
119+
concurrency:
120+
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ inputs.repository }}-${{ github.event_name == 'workflow_dispatch' }}
121+
cancel-in-progress: true

requirements.txt

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
black
22
click
33
cmake
4-
fbgemm-gpu
4+
fbgemm-gpu>=1.4.0
55
hypothesis==6.70.1
66
importlib-metadata
77
iopath
@@ -16,7 +16,5 @@ tqdm
1616
usort
1717
parameterized
1818
PyYAML
19-
20-
# for tests
21-
# https://github.com/pytorch/pytorch/blob/b96b1e8cff029bb0a73283e6e7f6cc240313f1dc/requirements.txt#L3
22-
expecttest
19+
psutil
20+
expecttest>=0.3.0

0 commit comments

Comments
 (0)