diff --git a/.ci/docker/requirements-ci.txt b/.ci/docker/requirements-ci.txt index aae483a7748..c624fc5de82 100644 --- a/.ci/docker/requirements-ci.txt +++ b/.ci/docker/requirements-ci.txt @@ -379,3 +379,6 @@ dataclasses_json==0.6.7 cmake==4.0.0 #Description: required for building + +tlparse==0.3.30 +#Description: required for log parsing diff --git a/.ci/pytorch/common_utils.sh b/.ci/pytorch/common_utils.sh index de2449d47c8..9c0e5242f43 100644 --- a/.ci/pytorch/common_utils.sh +++ b/.ci/pytorch/common_utils.sh @@ -159,11 +159,6 @@ function install_torchvision() { fi } -function install_tlparse() { - pip_install --user "tlparse==0.3.30" - PATH="$(python -m site --user-base)/bin:$PATH" -} - function install_torchrec_and_fbgemm() { local torchrec_commit torchrec_commit=$(get_pinned_commit torchrec) diff --git a/.ci/pytorch/macos-test.sh b/.ci/pytorch/macos-test.sh index a6c9d652475..81bc2cd4536 100755 --- a/.ci/pytorch/macos-test.sh +++ b/.ci/pytorch/macos-test.sh @@ -317,8 +317,6 @@ test_timm_perf() { echo "timm benchmark on mps device completed" } -install_tlparse - if [[ $TEST_CONFIG == *"perf_all"* ]]; then test_torchbench_perf test_hf_perf diff --git a/.ci/pytorch/test.sh b/.ci/pytorch/test.sh index dcaf8a87869..ad7f5872b30 100755 --- a/.ci/pytorch/test.sh +++ b/.ci/pytorch/test.sh @@ -212,8 +212,6 @@ if [[ "$BUILD_ENVIRONMENT" == *aarch64* ]]; then export VALGRIND=OFF fi -install_tlparse - # DANGER WILL ROBINSON. The LD_PRELOAD here could cause you problems # if you're not careful. Check this if you made some changes and the # ASAN test is not working diff --git a/.github/requirements/pip-requirements-macOS.txt b/.github/requirements/pip-requirements-macOS.txt index 2bb099373c6..1b08af886f8 100644 --- a/.github/requirements/pip-requirements-macOS.txt +++ b/.github/requirements/pip-requirements-macOS.txt @@ -27,6 +27,7 @@ pytest==7.3.2 pyyaml==6.0.2 scipy==1.12.0 sympy==1.13.3 +tlparse==0.3.30 tensorboard==2.13.0 typing-extensions==4.12.2 unittest-xml-reporting<=3.2.0,>=2.0.0