diff --git a/etgtools/pi_generator.py b/etgtools/pi_generator.py index 1ee4a130..b1507d0a 100644 --- a/etgtools/pi_generator.py +++ b/etgtools/pi_generator.py @@ -82,7 +82,7 @@ typing_imports = """\ from __future__ import annotations from collections.abc import Callable from enum import IntEnum, IntFlag, auto -from typing import Any, overload, TypeAlias, TypeVar, ParamSpec, Generic, Union, Optional +from typing import Any, overload, TypeAlias, TypeVar, ParamSpec, Generic, Union, Optional, List """ diff --git a/etgtools/tweaker_tools.py b/etgtools/tweaker_tools.py index 32f7fef7..2235426d 100644 --- a/etgtools/tweaker_tools.py +++ b/etgtools/tweaker_tools.py @@ -213,11 +213,11 @@ class FixWxPrefix(object): if type_name.startswith('Vector<') and type_name.endswith('>'): # Special handling for 'Vector' types type_name = self.cleanType(type_name[7:-1]) - return f'list[{type_name}]' + return f'List[{type_name}]' if type_name.startswith('Array'): type_name = self.cleanType(type_name[5:]) if type_name: - return f'list[{type_name}]' + return f'List[{type_name}]' else: return 'list' return type_map.get(type_name, type_name)