incorporating code review comments from Daniel

lokeshrangineni · lokeshrangineni · commit 7e324aeaf4d4 · 2024-06-12T08:58:40.000-04:00
Signed-off-by: Lokesh Rangineni &lt;lokeshforjava@gmail.com&gt;

Signed-off-by: Lokesh Rangineni &lt;lokeshforjava@gmail.com&gt;
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,
