From fcfd52a80b7817434cb080202a642ea2a5be6c5d Mon Sep 17 00:00:00 2001 From: Jelle Aalbers Date: Mon, 31 Jul 2023 17:07:44 -0700 Subject: [PATCH] Fix array_columns type signature --- flamedisx/source.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flamedisx/source.py b/flamedisx/source.py index 0bbcd468c..030b17178 100644 --- 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()