isort linting

Signed-off-by: Chester Ong <chester.ong.ch@gmail.com>
feast-dev · bushwhackr · Feb 11, 2024 · Feb 11, 2024 · Feb 11, 2024 · Feb 11, 2024
commit a96c266b575f1895f1a6727b9df1af97700402b4
@@ -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

@@ -1,6 +1,6 @@
 import os
 import uuid
-from typing import Dict, List, Optional
+from typing import List
 
 import pandas as pd
 

@@ -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,

@@ -1,5 +1,5 @@
 import logging
-from typing import Dict, Optional
+from typing import Dict
 
 import pandas as pd
 import pytest

@@ -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

@@ -1,6 +1,6 @@
 import pathlib
 import uuid
-from typing import Dict, List, Optional
+from typing import List
 
 import pandas as pd
 import pytest

@@ -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,

@@ -1,5 +1,4 @@
 from abc import ABC, abstractmethod
-from typing import Dict, Optional
 
 import pandas as pd
 

@@ -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

@@ -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

@@ -1,6 +1,6 @@
 import os
 import uuid
-from typing import Dict, List, Optional
+from typing import List
 
 import pandas as pd
 

@@ -1,6 +1,6 @@
 import os
 import uuid
-from typing import Dict, List, Optional
+from typing import List
 
 import pandas as pd
 

@@ -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

@@ -1,5 +1,4 @@
 import os
-from typing import Dict
 
 from google.cloud import datastore
 from testcontainers.core.container import DockerContainer

@@ -1,5 +1,3 @@
-from typing import Dict
-
 from testcontainers.core.container import DockerContainer
 from testcontainers.core.waiting_utils import wait_for_logs
 

@@ -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

@@ -1,5 +1,3 @@
-from typing import Dict
-
 from testcontainers.core.container import DockerContainer
 from testcontainers.core.waiting_utils import wait_for_logs
 

@@ -1,5 +1,3 @@
-from typing import Dict
-
 from testcontainers.mysql import MySqlContainer
 
 from feast.repo_config import FeastConfigBaseModel

@@ -1,5 +1,3 @@
-from typing import Dict
-
 from testcontainers.core.container import DockerContainer
 from testcontainers.core.waiting_utils import wait_for_logs