use workspace param

Author: BrianPei

.github/workflows/unit_tests_common.yml

@@ -201,10 +201,11 @@ jobs:
 
             echo "===== Step 7: Install TE-FL Plugin Layer ====="
             # Install TransformerEngine-FL Python layer (plugin logic)
-            cd /workspace/TransformerEngine-FL
+            cd $GITHUB_WORKSPACE
             TE_FL_SKIP_CUDA=1 python3 setup.py install
 
             echo "===== Step 8: Final Verification ====="
+            python3 tests/pytorch/test_sanity_import.py
             # Verify both TE Python API and backend are functional
             python3 -c "import transformer_engine; import transformer_engine_torch as te; print('transformer_engine:', transformer_engine); print('transformer_engine_torch:', te)"