Use the python path from virtualenv in end to end tests

Signed-off-by: hemildesai <[email protected]>

Author: hemildesai

end-to-end-test/end_to_end_test/test_checkout.py

@@ -5,7 +5,8 @@
 import os
 import subprocess
 import pytest  # type: ignore
-from .utils import get_env
+
+from .utils import get_env, PYTHON_PATH
 
 
 @pytest.mark.parametrize(
@@ -69,7 +70,7 @@ def main():
         )
 
     env = get_env()
-    cmd = ["python", "train.py"]
+    cmd = [PYTHON_PATH, "train.py"]
     subprocess.run(cmd, cwd=tmpdir, env=env, check=True)
 
     experiments = json.loads(

end-to-end-test/end_to_end_test/test_delete.py

@@ -4,7 +4,7 @@
 from pathlib import Path
 import pytest  # type: ignore
 
-from .utils import path_exists, get_env
+from .utils import path_exists, get_env, PYTHON_PATH
 
 
 @pytest.mark.parametrize(
@@ -57,7 +57,7 @@ def main():
         )
 
     env = get_env()
-    cmd = ["python", "train.py", "--foo"]
+    cmd = [PYTHON_PATH, "train.py", "--foo"]
     subprocess.run(cmd, cwd=tmpdir, env=env, check=True)
 
     experiments = json.loads(

end-to-end-test/end_to_end_test/test_list.py

@@ -4,7 +4,7 @@
 import pytest  # type: ignore
 from dateutil.parser import parse as parse_date
 
-from .utils import get_env
+from .utils import get_env, PYTHON_PATH
 
 
 @pytest.mark.parametrize(
@@ -58,7 +58,7 @@ def main():
 
     env = get_env()
 
-    subprocess.run(["python", "train.py", "--foo"], cwd=tmpdir, env=env, check=True)
+    subprocess.run([PYTHON_PATH, "train.py", "--foo"], cwd=tmpdir, env=env, check=True)
 
     experiments = json.loads(
         subprocess.run(
@@ -106,7 +106,9 @@ def main():
 experiment.checkpoint(path=".", step=3)
 """
         )
-    subprocess.run(["python", "train2.py", exp["id"]], cwd=tmpdir, env=env, check=True)
+    subprocess.run(
+        [PYTHON_PATH, "train2.py", exp["id"]], cwd=tmpdir, env=env, check=True
+    )
     experiments = json.loads(
         subprocess.run(
             ["replicate", "--verbose", "list", "--json"],

end-to-end-test/end_to_end_test/utils.py

@@ -5,6 +5,7 @@
 from google.cloud import storage as google_storage
 
 ROOT_DIRECTORY = Path(__file__).parent.parent.parent
+PYTHON_PATH = os.path.abspath(os.path.join(ROOT_DIRECTORY, "venv/bin/python"))
 
 
 def get_env():