freeze tensorflow dep

voxel51 · Oct 28, 2024 · db6a02a · db6a02a
1 parent 43411ee
commit db6a02a
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