From 603153a4c7cd38625d8be9ae1fdad22678e68b17 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Sat, 6 Jun 2026 07:52:40 +0000
Subject: [PATCH 1/2] Initial plan


From 2c0e17e6b51792feae542032237bf1bee1c1b189 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Sat, 6 Jun 2026 07:55:45 +0000
Subject: [PATCH 2/2] Fix torch_scatter ABI mismatch in CI test script

---
 ut.sh | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/ut.sh b/ut.sh
index ba53ffe..e18a477 100644
--- a/ut.sh
+++ b/ut.sh
@@ -13,7 +13,15 @@ echo "--- Installing/updating package from PR in editable mode ---"
 #    Docker image.
 # 2. `pytest ./tests/`: After the package is installed, we run the tests.
 
-conda run -n dpnegf bash -c "pip install -e . && pytest dpnegf/tests/"
+conda run -n dpnegf bash -c "
+  pip install -e .
 
-echo "--- Unit Tests Passed Successfully ---"
+  TORCH_VERSION=\$(python -c \"import torch; print(torch.__version__.split('+')[0])\")
+  TORCH_BACKEND=\$(python -c \"import torch; print('cpu' if torch.version.cuda is None else 'cu' + torch.version.cuda.replace('.', ''))\")
+  TORCH_SCATTER_INDEX=\"https://data.pyg.org/whl/torch-\${TORCH_VERSION}+\${TORCH_BACKEND}.html\"
+  pip install --no-cache-dir --force-reinstall --only-binary=torch_scatter torch_scatter -f \"\${TORCH_SCATTER_INDEX}\"
+
+  pytest dpnegf/tests/
+"
 
+echo "--- Unit Tests Passed Successfully ---"