diff --git a/sdk/python/feast/infra/contrib/spark_kafka_processor.py b/sdk/python/feast/infra/contrib/spark_kafka_processor.py index 1c1fb07c41..829b43ee48 100644 --- a/sdk/python/feast/infra/contrib/spark_kafka_processor.py +++ b/sdk/python/feast/infra/contrib/spark_kafka_processor.py @@ -5,7 +5,6 @@ from pyspark.sql import DataFrame, SparkSession from pyspark.sql.avro.functions import from_avro from pyspark.sql.functions import col, from_json -from pyspark.sql.streaming import StreamingQuery from feast.data_format import AvroFormat, JsonFormat from feast.data_source import KafkaSource, PushMode diff --git a/sdk/python/feast/infra/offline_stores/contrib/athena_offline_store/tests/data_source.py b/sdk/python/feast/infra/offline_stores/contrib/athena_offline_store/tests/data_source.py index 4d16e2a9a5..4d29cf064d 100644 --- a/sdk/python/feast/infra/offline_stores/contrib/athena_offline_store/tests/data_source.py +++ b/sdk/python/feast/infra/offline_stores/contrib/athena_offline_store/tests/data_source.py @@ -1,6 +1,6 @@ import os import uuid -from typing import Dict, List, Optional +from typing import List import pandas as pd diff --git a/sdk/python/feast/infra/offline_stores/contrib/mssql_offline_store/tests/data_source.py b/sdk/python/feast/infra/offline_stores/contrib/mssql_offline_store/tests/data_source.py index 8507414da6..04efbbb35b 100644 --- a/sdk/python/feast/infra/offline_stores/contrib/mssql_offline_store/tests/data_source.py +++ b/sdk/python/feast/infra/offline_stores/contrib/mssql_offline_store/tests/data_source.py @@ -1,4 +1,4 @@ -from typing import Dict, List, Optional +from typing import List import pandas as pd import pytest @@ -14,7 +14,6 @@ from feast.infra.offline_stores.contrib.mssql_offline_store.mssqlserver_source import ( MsSqlServerSource, ) -from feast.repo_config import FeastConfigBaseModel from feast.saved_dataset import SavedDatasetStorage from tests.integration.feature_repos.universal.data_source_creator import ( DataSourceCreator, diff --git a/sdk/python/feast/infra/offline_stores/contrib/postgres_offline_store/tests/data_source.py b/sdk/python/feast/infra/offline_stores/contrib/postgres_offline_store/tests/data_source.py index c9d0ac0060..1e378b7a25 100644 --- a/sdk/python/feast/infra/offline_stores/contrib/postgres_offline_store/tests/data_source.py +++ b/sdk/python/feast/infra/offline_stores/contrib/postgres_offline_store/tests/data_source.py @@ -1,5 +1,5 @@ import logging -from typing import Dict, Optional +from typing import Dict import pandas as pd import pytest diff --git a/sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/tests/data_source.py b/sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/tests/data_source.py index f0cb5a7bd9..d14d07de85 100644 --- a/sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/tests/data_source.py +++ b/sdk/python/feast/infra/offline_stores/contrib/spark_offline_store/tests/data_source.py @@ -2,7 +2,7 @@ import shutil import tempfile import uuid -from typing import Dict, List, Optional +from typing import List import pandas as pd from pyspark import SparkConf diff --git a/sdk/python/feast/infra/offline_stores/contrib/trino_offline_store/tests/data_source.py b/sdk/python/feast/infra/offline_stores/contrib/trino_offline_store/tests/data_source.py index 162e12cbd4..850a596586 100644 --- a/sdk/python/feast/infra/offline_stores/contrib/trino_offline_store/tests/data_source.py +++ b/sdk/python/feast/infra/offline_stores/contrib/trino_offline_store/tests/data_source.py @@ -1,6 +1,6 @@ import pathlib import uuid -from typing import Dict, List, Optional +from typing import List import pandas as pd import pytest diff --git a/sdk/python/feast/infra/offline_stores/snowflake.py b/sdk/python/feast/infra/offline_stores/snowflake.py index 9446ead3df..088f0105f8 100644 --- a/sdk/python/feast/infra/offline_stores/snowflake.py +++ b/sdk/python/feast/infra/offline_stores/snowflake.py @@ -1,6 +1,5 @@ import contextlib import json -import os import uuid import warnings from datetime import datetime @@ -23,7 +22,6 @@ import numpy as np import pandas as pd import pyarrow -from pydantic import Field, StrictStr from pydantic.typing import Literal from pytz import utc @@ -51,7 +49,7 @@ write_pandas, write_parquet, ) -from feast.repo_config import FeastConfigBaseModel, RepoConfig +from feast.repo_config import RepoConfig from feast.saved_dataset import SavedDatasetStorage from feast.types import ( Array, diff --git a/sdk/python/tests/integration/feature_repos/universal/data_source_creator.py b/sdk/python/tests/integration/feature_repos/universal/data_source_creator.py index 95419409fe..3e2d14f5c3 100644 --- a/sdk/python/tests/integration/feature_repos/universal/data_source_creator.py +++ b/sdk/python/tests/integration/feature_repos/universal/data_source_creator.py @@ -1,5 +1,4 @@ from abc import ABC, abstractmethod -from typing import Dict, Optional import pandas as pd diff --git a/sdk/python/tests/integration/feature_repos/universal/data_sources/bigquery.py b/sdk/python/tests/integration/feature_repos/universal/data_sources/bigquery.py index 7464f28103..2a0e0bc9f6 100644 --- a/sdk/python/tests/integration/feature_repos/universal/data_sources/bigquery.py +++ b/sdk/python/tests/integration/feature_repos/universal/data_sources/bigquery.py @@ -1,6 +1,6 @@ import os import uuid -from typing import Dict, List, Optional +from typing import List, Optional import pandas as pd from google.cloud import bigquery diff --git a/sdk/python/tests/integration/feature_repos/universal/data_sources/file.py b/sdk/python/tests/integration/feature_repos/universal/data_sources/file.py index 6e51bdc757..aad8543224 100644 --- a/sdk/python/tests/integration/feature_repos/universal/data_sources/file.py +++ b/sdk/python/tests/integration/feature_repos/universal/data_sources/file.py @@ -2,7 +2,7 @@ import shutil import tempfile import uuid -from typing import Any, Dict, List, Optional +from typing import Any, List import pandas as pd import pyarrow as pa diff --git a/sdk/python/tests/integration/feature_repos/universal/data_sources/redshift.py b/sdk/python/tests/integration/feature_repos/universal/data_sources/redshift.py index c400fac69b..eb65e9b53c 100644 --- a/sdk/python/tests/integration/feature_repos/universal/data_sources/redshift.py +++ b/sdk/python/tests/integration/feature_repos/universal/data_sources/redshift.py @@ -1,6 +1,6 @@ import os import uuid -from typing import Dict, List, Optional +from typing import List import pandas as pd diff --git a/sdk/python/tests/integration/feature_repos/universal/data_sources/snowflake.py b/sdk/python/tests/integration/feature_repos/universal/data_sources/snowflake.py index 0d07fdaaf3..a318b2a253 100644 --- a/sdk/python/tests/integration/feature_repos/universal/data_sources/snowflake.py +++ b/sdk/python/tests/integration/feature_repos/universal/data_sources/snowflake.py @@ -1,6 +1,6 @@ import os import uuid -from typing import Dict, List, Optional +from typing import List import pandas as pd diff --git a/sdk/python/tests/integration/feature_repos/universal/online_store/cassandra.py b/sdk/python/tests/integration/feature_repos/universal/online_store/cassandra.py index da1fbcb40f..13dac69397 100644 --- a/sdk/python/tests/integration/feature_repos/universal/online_store/cassandra.py +++ b/sdk/python/tests/integration/feature_repos/universal/online_store/cassandra.py @@ -15,7 +15,6 @@ # import time -from typing import Dict from testcontainers.core.container import DockerContainer from testcontainers.core.waiting_utils import wait_for_logs diff --git a/sdk/python/tests/integration/feature_repos/universal/online_store/datastore.py b/sdk/python/tests/integration/feature_repos/universal/online_store/datastore.py index 26954ca0fe..52d0e8243a 100644 --- a/sdk/python/tests/integration/feature_repos/universal/online_store/datastore.py +++ b/sdk/python/tests/integration/feature_repos/universal/online_store/datastore.py @@ -1,5 +1,4 @@ import os -from typing import Dict from google.cloud import datastore from testcontainers.core.container import DockerContainer diff --git a/sdk/python/tests/integration/feature_repos/universal/online_store/dynamodb.py b/sdk/python/tests/integration/feature_repos/universal/online_store/dynamodb.py index 8445d5393f..35781cccb1 100644 --- a/sdk/python/tests/integration/feature_repos/universal/online_store/dynamodb.py +++ b/sdk/python/tests/integration/feature_repos/universal/online_store/dynamodb.py @@ -1,5 +1,3 @@ -from typing import Dict - from testcontainers.core.container import DockerContainer from testcontainers.core.waiting_utils import wait_for_logs diff --git a/sdk/python/tests/integration/feature_repos/universal/online_store/hazelcast.py b/sdk/python/tests/integration/feature_repos/universal/online_store/hazelcast.py index 6d3624f715..233021e185 100644 --- a/sdk/python/tests/integration/feature_repos/universal/online_store/hazelcast.py +++ b/sdk/python/tests/integration/feature_repos/universal/online_store/hazelcast.py @@ -1,7 +1,6 @@ import logging import random import string -from typing import Any, Dict from testcontainers.core.container import DockerContainer from testcontainers.core.waiting_utils import wait_for_logs diff --git a/sdk/python/tests/integration/feature_repos/universal/online_store/hbase.py b/sdk/python/tests/integration/feature_repos/universal/online_store/hbase.py index 461e29b173..430a54c28c 100644 --- a/sdk/python/tests/integration/feature_repos/universal/online_store/hbase.py +++ b/sdk/python/tests/integration/feature_repos/universal/online_store/hbase.py @@ -1,5 +1,3 @@ -from typing import Dict - from testcontainers.core.container import DockerContainer from testcontainers.core.waiting_utils import wait_for_logs diff --git a/sdk/python/tests/integration/feature_repos/universal/online_store/mysql.py b/sdk/python/tests/integration/feature_repos/universal/online_store/mysql.py index f0ee61a04f..6c54c0119e 100644 --- a/sdk/python/tests/integration/feature_repos/universal/online_store/mysql.py +++ b/sdk/python/tests/integration/feature_repos/universal/online_store/mysql.py @@ -1,5 +1,3 @@ -from typing import Dict - from testcontainers.mysql import MySqlContainer from feast.repo_config import FeastConfigBaseModel diff --git a/sdk/python/tests/integration/feature_repos/universal/online_store/redis.py b/sdk/python/tests/integration/feature_repos/universal/online_store/redis.py index cc27a83ca7..62e8b28262 100644 --- a/sdk/python/tests/integration/feature_repos/universal/online_store/redis.py +++ b/sdk/python/tests/integration/feature_repos/universal/online_store/redis.py @@ -1,5 +1,3 @@ -from typing import Dict - from testcontainers.core.container import DockerContainer from testcontainers.core.waiting_utils import wait_for_logs