incorporating code review comments from Daniel

Signed-off-by: Lokesh Rangineni <lokeshforjava@gmail.com> Signed-off-by: Lokesh Rangineni <lokeshforjava@gmail.com>
feast-dev · Jun 12, 2024 · 7e324ae · 7e324ae
1 parent 8ee2df3
commit 7e324ae
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/sdk/python/tests/integration/feature_repos/repo_configuration.py b/sdk/python/tests/integration/feature_repos/repo_configuration.py
@@ -465,6 +465,13 @@ def construct_test_environment(
         project, fixture_request=fixture_request
     )
 
+    if test_repo_config.online_store_creator:
+        online_creator = test_repo_config.online_store_creator(
+            project, fixture_request=fixture_request
+        )
+    else:
+        online_creator = None
+
     if test_repo_config.python_feature_server and test_repo_config.provider == "aws":
         from feast.infra.feature_servers.aws_lambda.config import (
             AwsLambdaFeatureServerConfig,
@@ -498,13 +505,6 @@ def construct_test_environment(
             cache_ttl_seconds=1,
         )
 
-    if test_repo_config.online_store_creator:
-        online_creator = test_repo_config.online_store_creator(
-            project, fixture_request=fixture_request, registry_path=registry.path
-        )
-    else:
-        online_creator = None
-
     environment = Environment(
         name=project,
         provider=test_repo_config.provider,