diff --git a/tests/test_graph_laplacian.py b/tests/test_graph_laplacian.py index 70fa65b..6f7737d 100644 --- a/tests/test_graph_laplacian.py +++ b/tests/test_graph_laplacian.py @@ -4,9 +4,7 @@ import torch from scipy.sparse import lil_matrix, csr_matrix, coo_matrix import os import sys -#maindir = os.getcwd() -#main_path = maindir[:maindir.find('ds4ms/code')] -#sys.path.append(main_path+"/ds4ms/code/src") + from AMBER.graph_laplacian import create_laplacian_matrix, delete_from_csr, \ remove_vertex, laplacian, unnormalized_laplacian, laplacian_chain, \ unnormalized_laplacian_chain