fix formatting

tokoko · tokoko · commit e88147470aa5 · 2024-05-25T07:07:56.000Z
Signed-off-by: tokoko &lt;togurg14@freeuni.edu.ge&gt;
diff --git a/sdk/python/tests/integration/registration/test_feature_store.py b/sdk/python/tests/integration/registration/test_feature_store.py
@@ -11,25 +11,18 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import os
-import time
 from datetime import timedelta
 from tempfile import mkstemp
 
 import pytest
 from pytest_lazyfixture import lazy_fixture
 
-from feast import FileSource
-from feast.data_format import ParquetFormat
 from feast.entity import Entity
 from feast.feature_store import FeatureStore
 from feast.feature_view import FeatureView
-from feast.field import Field
-from feast.infra.offline_stores.file import FileOfflineStoreConfig
-from feast.infra.online_stores.dynamodb import DynamoDBOnlineStoreConfig
 from feast.infra.online_stores.sqlite import SqliteOnlineStoreConfig
 from feast.repo_config import RepoConfig
-from feast.types import Array, Bytes, Float64, Int64, String
+from feast.types import Float64, Int64, String
 from tests.utils.data_source_test_creator import prep_file_source
 
 
@@ -68,6 +61,7 @@ def test_feature_view_inference_success(test_feature_store, dataframe_source):
 
         test_feature_store.teardown()
 
+
 @pytest.fixture
 def feature_store_with_local_registry():
     fd, registry_path = mkstemp()
@@ -81,5 +75,3 @@ def feature_store_with_local_registry():
             entity_key_serialization_version=2,
         )
     )
-
-
diff --git a/sdk/python/tests/integration/registration/test_universal_registry.py b/sdk/python/tests/integration/registration/test_universal_registry.py
@@ -232,33 +232,29 @@ def mock_remote_registry():
     registry.stub = RegistryServer_pb2_grpc.RegistryServerStub(mock_channel)
     yield registry
 
+
 if os.getenv("FEAST_IS_LOCAL_TEST", "False") == "False":
-    all_fixtures=[
-        "s3_registry", 
-        "gcs_registry"
-    ]
+    all_fixtures = ["s3_registry", "gcs_registry"]
 else:
     all_fixtures = [
-        "local_registry", 
-        "minio_registry", 
-        "pg_registry", 
-        "mysql_registry", 
-        "sqlite_registry", 
-        "mock_remote_registry"
+        "local_registry",
+        "minio_registry",
+        "pg_registry",
+        "mysql_registry",
+        "sqlite_registry",
+        "mock_remote_registry",
     ]
 
 
 # sql_fixtures = [
-#     "pg_registry", 
-#     "mysql_registry", 
-#     "sqlite_registry", 
+#     "pg_registry",
+#     "mysql_registry",
+#     "sqlite_registry",
 # ]
 
+
 @pytest.mark.integration
-@pytest.mark.parametrize(
-    "test_registry",
-    [lazy_fixture(f) for f in all_fixtures]
-)
+@pytest.mark.parametrize("test_registry", [lazy_fixture(f) for f in all_fixtures])
 def test_apply_entity_success(test_registry):
     entity = Entity(
         name="driver_car_id",
