Merge pull request #2348 from AI-Hypercomputer:check_correct_py_env

PiperOrigin-RevId: 807395598

Author: Google-ML-Automation

src/install_maxtext_extra_deps/install_github_deps.py

@@ -18,29 +18,25 @@
 
 def main():
     """
-    Installs extra dependencies specified in extra_deps.txt using uv.
+    Installs extra dependencies specified in extra_deps_from_github.txt using uv.
 
-    This script looks for 'extra_deps.txt' relative to its own location.
+    This script looks for 'extra_deps_from_github.txt' relative to its own location.
     It executes 'uv pip install -r <path_to_extra_deps.txt> --resolution=lowest'.
     """
     script_dir = Path(__file__).resolve().parent
 
-    # Adjust this path if your extra_deps.txt is in a different location,
+    # Adjust this path if your extra_deps_from_github.txt is in a different location,
     # e.g., script_dir / "data" / "extra_deps_from_github.txt"
     extra_deps_file = script_dir / "extra_deps_from_github.txt"
 
     if not extra_deps_file.exists():
         print(f"Error: '{extra_deps_file}' not found.")
-        print("Please ensure 'extra_deps.txt' is in the correct location relative to the script.")
+        print("Please ensure 'extra_deps_from_github.txt' is in the correct location relative to the script.")
         sys.exit(1)
-    # Check if 'uv' is available in the system's PATH
+    # Check if 'uv' is available in the environment
     try:
-        subprocess.run(["uv", "--version"], check=True, capture_output=True)
-    except FileNotFoundError:
-        print("Error: 'uv' command not found.")
-        subprocess.run(["pip", "install", "uv"], check=True)
-        subprocess.run(["uv", "--version"], check=True, capture_output=True)
-        sys.exit(1)
+        subprocess.run([sys.executable, "-m", "pip", "install", "uv"], check=True, capture_output=True)
+        subprocess.run([sys.executable, "-m", "uv", "--version"], check=True, capture_output=True)
     except subprocess.CalledProcessError as e:
         print(f"Error checking uv version: {e}")
         print(f"Stderr: {e.stderr.decode()}")