fix

Author: pathfinder-fp

.github/workflows/tpu-tests.yml

@@ -195,20 +195,20 @@ jobs:
         with:
           fetch-depth: 0
 
-#      - name: Setup Tunix , tpu-inference and dependencies
-#        run: |
-#          echo "Current directory:"
-#          pwd
-#          pip install --upgrade pip setuptools wheel
-#
-#          # Install Tunix
-#          pip uninstall torch torch-xla libtpu jax jaxlib -y
-#          pip install -e .[dev]
-#
-#          # Install tpu-inference
+      - name: Setup Tunix , tpu-inference and dependencies
+        run: |
+          echo "Current directory:"
+          pwd
+          pip install --upgrade pip setuptools wheel
+
+          # Install Tunix
+          pip uninstall torch torch-xla libtpu jax jaxlib -y
+          pip install -e .[dev]
+
+           Install tpu-inference
 #          pip uninstall torch libtpu jax jaxlib -y
 #          pip install tpu-inference==v0.11.1 --force-reinstall
-#          pip install pytest pytest-xdist
+          pip install pytest pytest-xdist
 
 #      - name: Run tests
 #        env: