Merge pull request #4995 from voxel51/fix/psutil-dep

Freeze psutil/tensorflow dep
voxel51 · Oct 28, 2024 · 8e51280 · 8e51280
2 parents 43411ee + db6a02a
commit 8e51280
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements/github.txt b/requirements/github.txt
@@ -4,7 +4,7 @@
 numpy<2
 pydicom>=2.2.0
 shapely>=1.7.1
-tensorflow
+tensorflow==2.17.0
 tensorflow-datasets
 torch
 torchvision
diff --git a/requirements/test.txt b/requirements/test.txt
@@ -7,5 +7,5 @@ pytest-cov==4.0.0
 pytest-mock==3.10.0
 pytest-asyncio
 shapely
-tensorflow
+tensorflow==2.17.0
 twine>=3