diff --git a/pythonfmu3/variables.py b/pythonfmu3/variables.py index c3aba08..c2ee44e 100644 --- a/pythonfmu3/variables.py +++ b/pythonfmu3/variables.py @@ -391,9 +391,10 @@ def to_xml(self) -> Element: return parent -class Enumeration(ModelVariable): - def __init__(self, name: str, start: Optional[Any] = None, declared_type: Optional[Any] = None, **kwargs): - super().__init__(name, **kwargs) +class Enumeration(ModelVariable, Arrayable): + def __init__(self, name: str, start: Optional[Any] = None, declared_type: Optional[Any] = None, dimensions: List[Dimension] = [], **kwargs): + ModelVariable.__init__(self, name, **kwargs) + Arrayable.__init__(self, dimensions, **kwargs) self.__attrs = {"start": start, "declaredType": declared_type} self._type = "Enumeration"