fix tests

PurdueAF · Feb 28, 2024 · 615909e · 615909e
1 parent d3c5191
commit 615909e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/af_benchmark/processor/uproot_processor.py b/af_benchmark/processor/uproot_processor.py
@@ -17,7 +17,7 @@ def get_column_list(self, file):
         if isinstance(columns_to_read, list):
             if any(c not in tree.keys() for c in columns_to_read):
                 raise ValueError(f"Error reading column: {column}")
-            self.column = columns_to_read
+            self.columns = columns_to_read
         elif isinstance(columns_to_read, int):
             self.columns = list(tree.keys())[:columns_to_read]
             if len(self.columns)<columns_to_read: