From 6c71105dc9622f755398e7f0cb534fea0c0ee1a9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lu=C3=ADs=20F=2E=20Pereira?= <luisfgper@gmail.com>
Date: Thu, 21 Nov 2024 16:26:54 -0800
Subject: [PATCH] Update env_setup to be compatible with pyproject with torch

---
 env_setup.sh | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/env_setup.sh b/env_setup.sh
index 41cbc9f9..325e0670 100755
--- a/env_setup.sh
+++ b/env_setup.sh
@@ -1,6 +1,4 @@
 #!/bin/bash -l
-pip install --upgrade pip
-pip install -e '.[all]'
 
 # Note that not all combinations of torch and CUDA are available
 # See https://github.com/pyg-team/pyg-lib to check the configuration that works for you
@@ -10,6 +8,7 @@ CUDA="cu118"    # if available, select the CUDA version suitable for your system
 pip install torch==${TORCH} --extra-index-url https://download.pytorch.org/whl/${CUDA}
 pip install pyg-lib torch-scatter torch-sparse torch-cluster -f https://data.pyg.org/whl/torch-${TORCH}+${CUDA}.html
 
-#pytest
+pip install --upgrade pip
+pip install -e '.[all]'
 
 pre-commit install