diff --git a/plugins/flytekit-omegaconf/flytekitplugins/omegaconf/dictconfig_transformer.py b/plugins/flytekit-omegaconf/flytekitplugins/omegaconf/dictconfig_transformer.py index e85ddb587d..d1b698cc76 100644 --- a/plugins/flytekit-omegaconf/flytekitplugins/omegaconf/dictconfig_transformer.py +++ b/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) diff --git a/plugins/flytekit-omegaconf/flytekitplugins/omegaconf/listconfig_transformer.py b/plugins/flytekit-omegaconf/flytekitplugins/omegaconf/listconfig_transformer.py index 0d3957fa26..b7fd641524 100644 --- a/plugins/flytekit-omegaconf/flytekitplugins/omegaconf/listconfig_transformer.py +++ b/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]): diff --git a/plugins/flytekit-omegaconf/flytekitplugins/omegaconf/type_information.py b/plugins/flytekit-omegaconf/flytekitplugins/omegaconf/type_information.py index a126d7f704..393c48c3b4 100644 --- a/plugins/flytekit-omegaconf/flytekitplugins/omegaconf/type_information.py +++ b/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)