Drop pip

Author: aIbrahiim

sdks/python/tox.ini

@@ -103,20 +103,25 @@ commands =
   python apache_beam/examples/complete/autocomplete_test.py
   bash {toxinidir}/scripts/run_pytest.sh {envname} "{posargs}"
 
-[testenv:py{310,311}-ml]
+[testenv:py310-ml]
 # Don't set TMPDIR to avoid "AF_UNIX path too long" errors in certain tests.
+install_command =
+  {envbindir}/python {envbindir}/pip install --retries 10 --constraint {toxinidir}/container/ml/py310/base_image_requirements.txt {opts} {packages}
+deps =
+pip_pre = False
+extras = test,gcp,dataframe,ml_test
+commands =
+  /bin/sh -c "pip freeze | grep -E tensorflow"
+  bash {toxinidir}/scripts/run_pytest.sh {envname} "{posargs}"
+
+[testenv:py311-ml]
+# Don't set TMPDIR to avoid "AF_UNIX path too long" errors in certain tests.
+install_command =
+  {envbindir}/python {envbindir}/pip install --retries 10 --constraint {toxinidir}/container/ml/py311/base_image_requirements.txt {opts} {packages}
 deps =
-  pip==25.0.1
-  accelerate>=1.6.0
-  onnx<1.19.0
-  google-cloud-aiplatform==1.138.0
-  googleapis-common-protos==1.72.0
-  grpcio-status==1.65.5
 pip_pre = False
-setenv =
 extras = test,gcp,dataframe,ml_test
 commands =
-  # Log tensorflow version for debugging
   /bin/sh -c "pip freeze | grep -E tensorflow"
   bash {toxinidir}/scripts/run_pytest.sh {envname} "{posargs}"