From 215c8889fdb323b923ab89b28f712a68929de637 Mon Sep 17 00:00:00 2001
From: "Haoyu (Daniel)" <yanghaoyu97@outlook.com>
Date: Fri, 2 Aug 2024 20:14:45 +0800
Subject: [PATCH] pin torch version until matgl release

---
 .github/workflows/test.yml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index b6d5930dd17..f6a9d5e99d6 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -75,9 +75,10 @@ jobs:
       - name: Install pymatgen and dependencies
         run: |
           micromamba activate pmg
-          # TODO remove temporary fix. added since uv install torch is flaky.
+          # TODO1: remove temporary fix. added since uv install torch is flaky.
           # track https://github.com/astral-sh/uv/issues/1921 for resolution
-          pip install torch --upgrade
+          # TODO2: remove touch version pin upon new matgl release
+          pip install torch==2.2.1 --upgrade
 
           uv pip install numpy cython
           uv pip install --editable '.[${{ matrix.config.extras }}]' --resolution=${{ matrix.config.resolution }}