fix test

agosztolai · May 22, 2024 · 0c9cb7b · 0c9cb7b
1 parent f848403
commit 0c9cb7b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/MARBLE/geometry.py b/MARBLE/geometry.py
@@ -365,11 +365,11 @@ def fit_graph(x, graph_type="cknn", par=1, delta=1.0, metric="euclidean"):
         edge_index = utils.np2torch(edge_index, dtype="double")
 
     elif graph_type == "knn":
-        edge_index = knn_graph(x, k=par, metric=metric)
+        edge_index = knn_graph(x, k=par)
         edge_index = PyGu.add_self_loops(edge_index)[0]
 
     elif graph_type == "radius":
-        edge_index = radius_graph(x, r=par, metric=metric)
+        edge_index = radius_graph(x, r=par)
         edge_index = PyGu.add_self_loops(edge_index)[0]
 
     else:

diff --git a/install.sh b/install.sh
@@ -4,7 +4,6 @@
 echo "Checking the PyTorch version"
 SED=$(which gsed || which sed)
 TORCH_VERSION=$(pip freeze | grep torch== | $SED -re "s/torch==([^+]+).*/\1/")
-echo $TORCH_VERSION
 
 if [ -z "$TORCH_VERSION" ]
 then