Unverified Commit 3d0bd869 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Install the tensorflow example requirements in docker (#31428)

parent 11f43c15
...@@ -155,7 +155,7 @@ class CircleCIJob: ...@@ -155,7 +155,7 @@ class CircleCIJob:
elif self.name in ["flax","torch","tf"]: elif self.name in ["flax","torch","tf"]:
name = self.name if self.name != "torch" else "" name = self.name if self.name != "torch" else ""
if self.name == "torch": if self.name == "torch":
all_tests = glob.glob(f"tests/models/**/test_modeling_{name}*.py", recursive=True) all_tests = glob.glob(f"tests/models/**/test_modeling_{name}*.py", recursive=True)
filtered = [k for k in all_tests if ("_tf_") not in k and "_flax_" not in k] filtered = [k for k in all_tests if ("_tf_") not in k and "_flax_" not in k]
expanded_tests.extend(filtered) expanded_tests.extend(filtered)
else: else:
...@@ -163,7 +163,7 @@ class CircleCIJob: ...@@ -163,7 +163,7 @@ class CircleCIJob:
else: else:
expanded_tests.extend(glob.glob("tests/models/**/test_modeling*.py", recursive=True)) expanded_tests.extend(glob.glob("tests/models/**/test_modeling*.py", recursive=True))
elif test == "tests/pipelines": elif test == "tests/pipelines":
expanded_tests.extend(glob.glob("tests/models/**/test_modeling*.py", recursive=True)) expanded_tests.extend(glob.glob("tests/models/**/test_modeling*.py", recursive=True))
else: else:
expanded_tests.append(test) expanded_tests.append(test)
tests = " ".join(expanded_tests) tests = " ".join(expanded_tests)
...@@ -326,7 +326,7 @@ examples_tensorflow_job = CircleCIJob( ...@@ -326,7 +326,7 @@ examples_tensorflow_job = CircleCIJob(
"examples_tensorflow", "examples_tensorflow",
cache_name="tensorflow_examples", cache_name="tensorflow_examples",
docker_image=[{"image":"huggingface/transformers-examples-tf"}], docker_image=[{"image":"huggingface/transformers-examples-tf"}],
install_steps=["uv venv && uv pip install ."], install_steps=["uv venv && uv pip install . && uv pip install -r examples/tensorflow/_tests_requirements.txt"],
parallelism=8 parallelism=8
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment