move FlyteSchema deprecation warning to initialization method (#1558)

* 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]>
flyteorg · Mar 23, 2023 · de28789 · de28789
1 parent a190431
commit de28789
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/flytekit/types/schema/types.py b/flytekit/types/schema/types.py
@@ -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]:
@@ -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
 
@@ -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 (