Add arguments to choose SPARE-* column names

CBICA · Aug 10, 2022 · 925469a · 925469a
1 parent 9dac338
commit 925469a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/NiBAx/core/model/datamodel.py b/NiBAx/core/model/datamodel.py
@@ -236,7 +236,8 @@ def GetDataStatistics(self):
 
         return stats
 
-    def AddSparesToDataModel(self,spares):
-        self.data.loc[:,'SPARE_AD'] = spares['SPARE_AD']
-        self.data.loc[:,'SPARE_BA'] = spares['SPARE_BA']
+    def AddSparesToDataModel(self,spares,SPARE_AD='SPARE_AD',
+                             SPARE_BA='SPARE_BA'):
+        self.data.loc[:,SPARE_AD] = spares['SPARE_AD']
+        self.data.loc[:,SPARE_BA] = spares['SPARE_BA']
         self.data_changed.emit()