Updated requirements

Enkhai · Jul 20, 2022 · 7d04fc7 · 7d04fc7
1 parent 308e3ea
commit 7d04fc7
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -12,11 +12,11 @@ torchaudio==0.11.0
 
 # PyTorch Geometric for the GNN
 # check setup dependency_links for links url
-torch-scatter==2.0.9
-torch-sparse==0.6.13
-torch-cluster==1.6.0
-torch-spline-conv==1.2.1
-torch-geometric==2.0.4
+torch-scatter
+torch-sparse
+torch-cluster
+torch-spline-conv
+torch-geometric
 
 # SB3
 stable-baselines3==1.5.0