SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Install the tensorflow example requirements in docker (#31428)

Browse Source
This commit is contained in:
amyeroberts
2024-06-14 19:35:43 +01:00
committed by GitHub
parent 11f43c15d3
commit 3d0bd86915
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
.circleci/create_circleci_config.py
View File

@@ -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:
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(
"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
) )