Update labels.py

vishesh9131 · Jun 7, 2024 · a1a66c8 · a1a66c8
1 parent ff58104
commit a1a66c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/USECASES/unittest/labels.py b/USECASES/unittest/labels.py
@@ -113,7 +113,7 @@ def setUp(self):
 adj_matrix = np.loadtxt(file_path, delimiter=",")
 
 # Read the CSV file into a DataFrame
-df = pd.read_csv("labelele.csv")
+df = pd.read_csv("./SANDBOX/labelele.csv")
 
 # # Find the top nodes
 top_nodes = find_top_nodes(adj_matrix, num_nodes=5)
@@ -126,7 +126,7 @@ class TestLabels(unittest.TestCase):
     def setUp(self):
         self.file_path = generate_random_graph(50, seed=122)
         self.adj_matrix = np.loadtxt(self.file_path, delimiter=",")
-        self.df = pd.read_csv("labelele.csv")
+        self.df = pd.read_csv("./SANDBOX/labelele.csv")
         self.col = self.df.values
         self.node_labels = {i: label for i, label in enumerate(self.col)}