add psutil in requirements.txt (#3601)

TroyGarden · facebook-github-bot · commit 43041c53f606 · 2025-12-09T09:48:07.000-08:00
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
diff --git a/.github/scripts/install_libs.sh b/.github/scripts/install_libs.sh
@@ -28,4 +28,4 @@ elif [ "$CHANNEL" = "test" ]; then
 fi
 
 
-${CONDA_RUN} pip install importlib-metadata click PyYAML
+${CONDA_RUN} pip install -r requirements.txt
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 black
 click
 cmake
-fbgemm-gpu
+fbgemm-gpu>=1.4.0
 hypothesis==6.70.1
 importlib-metadata
 iopath
@@ -16,7 +16,5 @@ tqdm
 usort
 parameterized
 PyYAML
-
-# for tests
-# https://github.com/pytorch/pytorch/blob/b96b1e8cff029bb0a73283e6e7f6cc240313f1dc/requirements.txt#L3
-expecttest
+psutil
+expecttest>=0.3.0

Original file line number	Diff line number	Diff line change
`@@ -28,4 +28,4 @@ elif [ "$CHANNEL" = "test" ]; then`
`28`	`28`	`fi`
`29`	`29`
`30`	`30`
`31`		`-${CONDA_RUN} pip install importlib-metadata click PyYAML`
	`31`	`+${CONDA_RUN} pip install -r requirements.txt`