Updated typing for transformation model

usc-isi-i2 · Nov 14, 2023 · 80d516f · 80d516f
1 parent e538629
commit 80d516f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sand/models/transformation.py b/sand/models/transformation.py
@@ -1,5 +1,5 @@
 from __future__ import annotations
-from typing import List, Union, Literal
+from typing import Literal
 from peewee import (
     CharField,
     ForeignKeyField,
@@ -17,8 +17,8 @@ class Transformation(BaseModel):
     name = CharField()
     table = ForeignKeyField(Table, backref="transformations", on_delete="CASCADE")
     mode = CharField()
-    datapath: Union[List[str], str] = JSONField()  # type: ignore
-    outputpath: List[str] = JSONField()  # type: ignore
+    datapath: list[str] | str = JSONField()  # type: ignore
+    outputpath: list[str] = JSONField()  # type: ignore
     type = CharField()
     code = TextField()
     on_error: Literal[