Skip to content

Commit

Permalink
move FlyteSchema deprecation warning to initialization method (#1558)
Browse files Browse the repository at this point in the history
* move FlyteSchema deprecation warning to initialization method

Signed-off-by: Niels Bantilan <[email protected]>

* update

Signed-off-by: Niels Bantilan <[email protected]>

---------

Signed-off-by: Niels Bantilan <[email protected]>
  • Loading branch information
cosmicBboy authored Mar 23, 2023
1 parent a190431 commit de28789
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion flytekit/types/schema/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -186,7 +186,6 @@ class FlyteSchema(object):
"""
This is the main schema class that users should use.
"""
logger.warning("FlyteSchema is deprecated, use Structured Dataset instead.")

@classmethod
def columns(cls) -> typing.Dict[str, typing.Type]:
Expand All @@ -203,6 +202,7 @@ def format(cls) -> SchemaFormat:
def __class_getitem__(
cls, columns: typing.Dict[str, typing.Type], fmt: SchemaFormat = SchemaFormat.PARQUET
) -> Type[FlyteSchema]:
logger.warning("FlyteSchema is deprecated, use Structured Dataset instead.")
if columns is None:
return FlyteSchema

Expand Down Expand Up @@ -240,6 +240,7 @@ def __init__(
supported_mode: SchemaOpenMode = SchemaOpenMode.WRITE,
downloader: typing.Optional[typing.Callable] = None,
):
logger.warning("FlyteSchema is deprecated, use Structured Dataset instead.")
if supported_mode == SchemaOpenMode.READ and remote_path is None:
raise ValueError("To create a FlyteSchema in read mode, remote_path is required")
if (
Expand Down

0 comments on commit de28789

Please sign in to comment.