Fix lint

Signed-off-by: Kevin Zhang <kzhang@tecton.ai>
feast-dev · feast-ci-bot · Aug 19, 2022 · Aug 5, 2022 · Aug 10, 2022 · Aug 10, 2022
commit 45f479ff7f260171db9b29f8473b30b171048c59
@@ -1,32 +1,18 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT license.
-import warnings
-from datetime import datetime, timedelta
-from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union
+from datetime import datetime
+from typing import Callable
 
-import pandas
-import pyarrow as pa
 from tqdm import tqdm
 
-from feast import FeatureService
-from feast.entity import Entity
-from feast.feature_logging import FeatureServiceLoggingSource
 from feast.feature_view import FeatureView
-from feast.infra.offline_stores.offline_store import RetrievalJob
-from feast.infra.offline_stores.offline_utils import get_offline_store_from_config
-from feast.infra.online_stores.helpers import get_online_store_from_config
 from feast.infra.passthrough_provider import PassthroughProvider
-from feast.infra.provider import Provider
-from feast.protos.feast.types.EntityKey_pb2 import EntityKey as EntityKeyProto
-from feast.protos.feast.types.Value_pb2 import Value as ValueProto
 from feast.infra.registry.base_registry import BaseRegistry
 from feast.repo_config import RepoConfig
-from feast.saved_dataset import SavedDataset
 from feast.utils import (
     _convert_arrow_to_proto,
     _get_column_names,
     _run_pyarrow_field_mapping,
-    make_tzaware,
 )
 
 DEFAULT_BATCH_SIZE = 10_000

@@ -30,8 +30,8 @@
     get_feature_view_query_context,
 )
 from feast.infra.provider import RetrievalJob
-from feast.on_demand_feature_view import OnDemandFeatureView
 from feast.infra.registry.base_registry import BaseRegistry
+from feast.on_demand_feature_view import OnDemandFeatureView
 from feast.repo_config import FeastBaseModel, RepoConfig
 from feast.saved_dataset import SavedDatasetStorage
 from feast.type_map import pa_to_mssql_type
@@ -178,7 +178,9 @@ def get_historical_features(
         engine = make_engine(config.offline_store)
         if isinstance(entity_df, pandas.DataFrame):
             entity_df_event_timestamp_col = (
-                offline_utils.infer_event_timestamp_from_entity_df(dict(zip(list(entity_df.columns), list(entity_df.dtypes))))
+                offline_utils.infer_event_timestamp_from_entity_df(
+                    dict(zip(list(entity_df.columns), list(entity_df.dtypes)))
+                )
             )
             entity_df[entity_df_event_timestamp_col] = pandas.to_datetime(
                 entity_df[entity_df_event_timestamp_col], utc=True

@@ -1,7 +1,6 @@
 from typing import Dict, List
 
 import pandas as pd
-import pyarrow as pa
 import pytest
 from sqlalchemy import create_engine
 from testcontainers.core.container import DockerContainer
@@ -10,7 +9,8 @@
 
 from feast.data_source import DataSource
 from feast.infra.offline_stores.contrib.mssql_offline_store.mssql import (
-    MsSqlServerOfflineStoreConfig, _df_to_create_table_sql
+    MsSqlServerOfflineStoreConfig,
+    _df_to_create_table_sql,
 )
 from feast.infra.offline_stores.contrib.mssql_offline_store.mssqlserver_source import (
     MsSqlServerSource,

diff --git a/sdk/python/feast/infra/registry_stores/contrib/azure/registry_store.py b/sdk/python/feast/infra/registry_stores/contrib/azure/registry_store.py
@@ -8,9 +8,9 @@
 from tempfile import TemporaryFile
 from urllib.parse import urlparse
 
-from feast.protos.feast.core.Registry_pb2 import Registry as RegistryProto
 from feast.infra.registry.registry import RegistryConfig
 from feast.infra.registry.registry_store import RegistryStore
+from feast.protos.feast.core.Registry_pb2 import Registry as RegistryProto
 
 REGISTRY_SCHEMA_VERSION = "1"