fix version problem

Signed-off-by: Future-Outlier <[email protected]>

.github/workflows/pythonbuild.yml

@@ -263,6 +263,7 @@ jobs:
           - flytekit-kf-pytorch
           - flytekit-kf-tensorflow
           - flytekit-mlflow
+          - flytekit-mmcloud
           - flytekit-modin
           - flytekit-onnx-pytorch
           - flytekit-onnx-scikitlearn

plugins/flytekit-pydantic/setup.py

@@ -4,7 +4,7 @@
 
 microlib_name = f"flytekitplugins-{PLUGIN_NAME}"
 
-plugin_requires = ["flytekit>=1.7.0,<2.0.0", "pydantic<2.0.0"]
+plugin_requires = ["flytekit>=1.7.0b0,<2.0.0", "pydantic"]
 
 __version__ = "0.0.0+develop"
 

plugins/flytekit-pydantic/tests/test_type_transformer.py

@@ -8,14 +8,19 @@
 from flyteidl.core.types_pb2 import SimpleType
 from flytekitplugins.pydantic import BaseModelTransformer
 from flytekitplugins.pydantic.commons import PYDANTIC_SUPPORTED_FLYTE_TYPES
-from pydantic import BaseModel, Extra
 
 import flytekit
 from flytekit.core import context_manager
 from flytekit.core.type_engine import TypeEngine
 from flytekit.types import directory
 from flytekit.types.file import file
 
+try:
+    # it will cause error if it is lesser than 2.0 version
+    from pydantic.v1 import BaseModel, Extra
+except ImportError:
+    from pydantic import BaseModel, Extra
+
 
 class TrainConfig(BaseModel):
     """Config BaseModel for testing purposes."""