check keras version to see why failing

.github/workflows/ci_tests.yml

@@ -29,7 +29,7 @@ jobs:
           pip install torch~=${{ matrix.TORCH_VER }}
           pip install keras coveralls pydot graphviz pytest pytest-cov
           pip install .
-          python -c "from astroNN.config import tf_patch; tf_patch()"
+          python -c "import keras; keras.__version__"
       - name: Cache test data
         uses: actions/cache@v3
         with:

.github/workflows/pypi.yml

@@ -20,9 +20,8 @@ jobs:
           python-version: "3.x"
       - name: Install dependencies
         run: |
-          pip install tensorflow
-          pip install tensorflow-probability
-          pip install coveralls pydot graphviz pytest pytest-cov build
+          pip install torch
+          pip install keras coveralls pydot graphviz pytest pytest-cov build
           pip install .
       - name: Build sdist
         run: |