diff options
Diffstat (limited to 'sqlglot/dataframe/sql/_typing.pyi')
-rw-r--r-- | sqlglot/dataframe/sql/_typing.pyi | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/sqlglot/dataframe/sql/_typing.pyi b/sqlglot/dataframe/sql/_typing.pyi index f1a03ea..67c8c09 100644 --- a/sqlglot/dataframe/sql/_typing.pyi +++ b/sqlglot/dataframe/sql/_typing.pyi @@ -10,11 +10,17 @@ if t.TYPE_CHECKING: from sqlglot.dataframe.sql.types import StructType ColumnLiterals = t.TypeVar( - "ColumnLiterals", bound=t.Union[str, float, int, bool, t.List, t.Tuple, datetime.date, datetime.datetime] + "ColumnLiterals", + bound=t.Union[str, float, int, bool, t.List, t.Tuple, datetime.date, datetime.datetime], ) ColumnOrName = t.TypeVar("ColumnOrName", bound=t.Union[Column, str]) ColumnOrLiteral = t.TypeVar( - "ColumnOrLiteral", bound=t.Union[Column, str, float, int, bool, t.List, t.Tuple, datetime.date, datetime.datetime] + "ColumnOrLiteral", + bound=t.Union[Column, str, float, int, bool, t.List, t.Tuple, datetime.date, datetime.datetime], +) +SchemaInput = t.TypeVar( + "SchemaInput", bound=t.Union[str, t.List[str], StructType, t.Dict[str, str]] +) +OutputExpressionContainer = t.TypeVar( + "OutputExpressionContainer", bound=t.Union[exp.Select, exp.Create, exp.Insert] ) -SchemaInput = t.TypeVar("SchemaInput", bound=t.Union[str, t.List[str], StructType, t.Dict[str, str]]) -OutputExpressionContainer = t.TypeVar("OutputExpressionContainer", bound=t.Union[exp.Select, exp.Create, exp.Insert]) |