uninstall all TE deps

Rohan-Bierneni · Rohan-Bierneni · commit 4e0dd7f1df2a · 2025-09-26T17:30:24.000Z
diff --git a/.github/workflows/UnitTests.yml b/.github/workflows/UnitTests.yml
@@ -37,7 +37,7 @@ jobs:
 
       - name: Print dependencies
         run: |
-          pip uninstall -y transformer-engine
+          pip uninstall -y transformer-engine transformer-engine-jax
           pip install transformer-engine
           pip freeze
 
