Fix array_columns type signature

FlamTeam · Mar 22, 2024 · fcfd52a · fcfd52a
1 parent 908d334
commit fcfd52a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/flamedisx/source.py b/flamedisx/source.py
@@ -46,8 +46,8 @@ class Source:
     #: Columns that we don't want to include in the tensor of data columns
     exclude_data_tensor: ty.Tuple[str] = tuple()
 
-    #: Names of array-valued data columns
-    array_columns: ty.Tuple[str] = tuple()
+    #: Array-valued data columns: tuple of (name, length)
+    array_columns: ty.Tuple[ty.Tuple[str, int]] = tuple()
 
     #: Any additional source attributes that should be configurable.
     model_attributes = tuple()