Don't override col_cls in DynamicTable.add_column

hdmf-dev · Apr 4, 2024 · a5f33c2 · a5f33c2
1 parent d85d0cb
commit a5f33c2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/hdmf/common/table.py b/src/hdmf/common/table.py
@@ -821,11 +821,13 @@ def add_column(self, **kwargs):  # noqa: C901
             raise ValueError("column '%s' cannot be both a table region "
                              "and come from an enumerable set of elements" % name)
         if table is not False:
-            col_cls = DynamicTableRegion
+            if col_cls is None:
+                 col_cls = DynamicTableRegion
             if isinstance(table, DynamicTable):
                 ckwargs['table'] = table
         if enum is not False:
-            col_cls = EnumData
+            if col_cls is None:
+                col_cls = EnumData
             if isinstance(enum, (list, tuple, np.ndarray, VectorData)):
                 ckwargs['elements'] = enum