diff --git a/convert.py b/convert.py index 5cc3f6e66..6c89b5ecc 100755 --- a/convert.py +++ b/convert.py @@ -530,7 +530,7 @@ class LazyTensor: raise ValueError(f'Cannot validate conversion from {self.data_type} to {data_type}.') -LazyModel = dict[str, LazyTensor] +LazyModel: TypeAlias = 'dict[str, LazyTensor]' @dataclass