Update connectivity.py

PennLINC · Oct 29, 2024 · 0c73fee · 0c73fee
1 parent 31ad69b
commit 0c73fee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xcp_d/interfaces/connectivity.py b/xcp_d/interfaces/connectivity.py
@@ -269,7 +269,7 @@ def flatten_conmat(df):
     df.columns = ["Row", "Column", "Value"]
     df["Edge"] = df["Row"] + "-" + df["Column"]
     df = df.set_index("Edge")
-    df = df[["Edge", "Value"]].T
+    df = df[["Value"]].T
     return df
 
 
@@ -857,7 +857,7 @@ def _run_interface(self, runtime):
             df.columns = ["Row", "Column", "Value"]
             df["Edge"] = df["Row"] + "-" + df["Column"]
             df = df.set_index("Edge")
-            df = df[["Edge", "Value"]].T
+            df = df[["Value"]].T
         elif self.inputs.kind == "labels":
             df = pd.read_table(in_file)
             df = pd.DataFrame(