Pin tensorflow due to unit tests

spacetelescope · Apr 3, 2024 · e5679cd · e5679cd
1 parent 5db73ee
commit e5679cd
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 10 deletions.
diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -17,12 +17,12 @@ jobs:
         uses: actions/checkout@v2
 
       - name: install requirements
-        run: | 
+        run: |
           pip install .[dev]
           pip install numpy scikit-learn tensorflow --upgrade-strategy only-if-needed
 
       - name: run pytest code coverage check
-        run: | 
+        run: |
           pytest \
             --verbose \
             --cov=calcloud \
@@ -54,9 +54,9 @@ jobs:
         uses: actions/checkout@v2
 
       - name: install requirements
-        run: | 
+        run: |
           pip install .[dev]
-          pip install numpy scikit-learn tensorflow --upgrade-strategy only-if-needed
+          pip install numpy scikit-learn==1.4.1 tensorflow==2.15.0 --upgrade-strategy only-if-needed
 
       - name: run pytest
         run: pytest -rP
diff --git a/calcloud/requirements.txt b/calcloud/requirements.txt
@@ -1,3 +1 @@
-tensorflow==2.12.1
-scikit-learn==1.3.0
 pyyaml
diff --git a/lambda/JobPredict/requirements.txt b/lambda/JobPredict/requirements.txt
@@ -1,4 +1,3 @@
-tensorflow==2.12.1
 scikit-learn==1.3.0
 boto3==1.28.56
 protobuf==4.24.3
diff --git a/lambda/ModelIngest/requirements.txt b/lambda/ModelIngest/requirements.txt
@@ -1,4 +1,2 @@
-tensorflow==2.12.1
-scikit-learn==1.3.0
 numpy
-boto3
+boto3