fix: Fix TRT-LLM container build when using a custom pip wheel (#1825)

container/build.sh

@@ -424,13 +424,15 @@ else
     fi
 fi
 
-cd "$NIXL_DIR" || exit
+pushd "$NIXL_DIR" || exit
 if ! git checkout ${NIXL_COMMIT}; then
     echo "ERROR: Failed to checkout NIXL commit ${NIXL_COMMIT}. The cached directory may be out of date."
     echo "Please delete $NIXL_DIR and re-run the build script."
     exit 1
 fi
 
+popd
+
 BUILD_CONTEXT_ARG+=" --build-context nixl=$NIXL_DIR"
 
 # Add NIXL_COMMIT as a build argument to enable caching