add pytest to consistency

.circleci/create_circleci_config.py

@@ -383,7 +383,7 @@ def job_name(self):
     docker_image=[{"image":"huggingface/transformers-consistency"}], # TODO do we really need torch for that job? torch-light?  
     install_steps=[
         "uv venv", "uv pip install -e .",
-        "sudo apt-get -y update && sudo apt-get install git-lfs",
+        "apt-get -y update && sudo apt-get install git-lfs",
         'git config --global user.email "[email protected]"',
         'git config --global user.name "ci"',
     ],
@@ -395,7 +395,7 @@ def job_name(self):
 onnx_job = CircleCIJob(
     "onnx",
     install_steps=[
-        "sudo apt-get -y update && sudo apt-get install -y cmake",
+        "apt-get -y update && apt-get install -y cmake",
         "pip install --upgrade --upgrade-strategy eager pip",
         "pip install -U --upgrade-strategy eager .[torch,tf,testing,sentencepiece,onnxruntime,vision,rjieba]",
     ],

docker/consistency.dockerfile

@@ -6,7 +6,7 @@ ENV VIRTUAL_ENV=/usr/local
 RUN pip install uv
 RUN uv venv
 RUN uv pip install --no-cache-dir -U pip setuptools
-RUN uv pip install --no-cache-dir tensorflow-cpu tf-keras
+RUN uv pip install --no-cache-dir tensorflow-cpu tf-keras pytest
 RUN uv pip install --no-cache --upgrade 'torch' --index-url https://download.pytorch.org/whl/cpu
 RUN pip install --no-cache-dir --upgrade GitPython "transformers[flax,quality]"
 RUN pip uninstall -y transformers