Skip to content

Commit

Permalink
move FlyteSchema deprecation warning to initialization method
Browse files Browse the repository at this point in the history
Signed-off-by: Niels Bantilan <[email protected]>
  • Loading branch information
cosmicBboy committed Mar 22, 2023
1 parent a190431 commit 3dfed17
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 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 Down Expand Up @@ -240,6 +239,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 3dfed17

Please sign in to comment.