Free up disk space in CI (zenml-io#1863)

* remove deepchecks tests from CI

* remove pytorch

* remove pytorch_lightning

* try tensorflow instead of pytorch

* formatting

* remove TF tests

* formatting

* try free ubuntu action

* refactor

* add skypilot back in

* install missing pkg and update tests

* add missing conditional to yaml

* with permissions

* remove tool cache as well

.github/workflows/setup-python-environment.yml

@@ -76,6 +76,27 @@ jobs:
     steps:
       - uses: actions/checkout@v3
 
+      - name: Free Disk Space (Ubuntu)
+        uses: jlumbroso/free-disk-space@main
+        if: ${{ inputs.os == 'ubuntu-latest' }}
+        with:
+          # this might remove tools that are actually needed,
+          # if set to "true" but frees about 6 GB
+          tool-cache: true
+
+          # all of these default to true, but feel free to set to
+          # "false" if necessary for your workflow
+          android: true
+          dotnet: true
+          haskell: true
+          large-packages: true
+          docker-images: true
+          swap-storage: true
+
+      - name: Reinstall packages needed for unit tests
+        if: ${{ inputs.os == 'ubuntu-latest' }}
+        run: sudo apt-get update && sudo apt-get install ffmpeg libsm6 libxext6 -y
+
       - name: Setup environment
         uses: ./.github/actions/setup_environment
         with:

tests/integration/examples/test_tensorflow.py

@@ -12,6 +12,7 @@
 #  or implied. See the License for the specific language governing
 #  permissions and limitations under the License.
 
+
 import sys
 
 import pytest