import flytekit logger instead of instantiating our own

Signed-off-by: mg515 <[email protected]>

plugins/flytekit-omegaconf/flytekitplugins/omegaconf/dictconfig_transformer.py

@@ -1,5 +1,4 @@
 import importlib
-import logging
 import re
 import typing
 from typing import Type, TypeVar
@@ -15,12 +14,11 @@
 from flytekit import FlyteContext
 from flytekit.core.type_engine import TypeTransformerFailedError
 from flytekit.extend import TypeEngine, TypeTransformer
+from flytekit.loggers import logger
 from flytekit.models.literals import Literal, Scalar
 from flytekit.models.types import LiteralType, SimpleType
 from omegaconf import DictConfig, OmegaConf
 
-logger = logging.getLogger("flytekit")
-
 T = TypeVar("T")
 NoneType = type(None)
 

plugins/flytekit-omegaconf/flytekitplugins/omegaconf/listconfig_transformer.py

@@ -1,5 +1,4 @@
 import importlib
-import logging
 from typing import Optional, Type, TypeVar
 
 from flyteidl.core.literals_pb2 import Literal as PB_Literal
@@ -17,7 +16,6 @@
 from omegaconf import ListConfig, OmegaConf
 
 T = TypeVar("T")
-logger = logging.getLogger("flytekit")
 
 
 class ListConfigTransformer(TypeTransformer[ListConfig]):

plugins/flytekit-omegaconf/flytekitplugins/omegaconf/type_information.py

@@ -1,14 +1,12 @@
 import dataclasses
-import logging
 import typing
 from collections import ChainMap
 
 from dataclasses_json import DataClassJsonMixin
 
+from flytekit.loggers import logger
 from omegaconf import DictConfig, ListConfig, OmegaConf
 
-logger = logging.getLogger("flytekit")
-
 NoneType = type(None)