From 31a39578de4355b28f56a5a581f9b8220126b8b1 Mon Sep 17 00:00:00 2001 From: Alex Sherstinsky Date: Tue, 16 Jan 2024 12:33:34 -0800 Subject: [PATCH] [BUGFIX] Pinning torch nightly to January 13, 2024 to avoid AttributeError (#3885) --- .github/workflows/pytest.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml index 1edbe4b0614..567b94861eb 100644 --- a/.github/workflows/pytest.yml +++ b/.github/workflows/pytest.yml @@ -126,7 +126,8 @@ jobs: if [ "$PYTORCH" == "nightly" ]; then extra_index_url=https://download.pytorch.org/whl/nightly/cpu - pip install --pre torch torchdata torchtext torchvision torchaudio --index-url $extra_index_url + # This is being tracked as the Issue "https://github.com/ludwig-ai/ludwig/issues/3886" in Ludwig GitHub repository. + pip install --pre torch==2.3.0.dev20240111+cpu torchtext torchvision torchaudio --extra-index-url $extra_index_url else extra_index_url=https://download.pytorch.org/whl/cpu