Fix tests

Signed-off-by: Kevin Zhang <kzhang@tecton.ai>
feast-dev · Mar 3, 2022 · 84fef3c · 84fef3c
1 parent dfad434
commit 84fef3c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sdk/python/tests/integration/registration/test_cli.py b/sdk/python/tests/integration/registration/test_cli.py
@@ -134,9 +134,11 @@ def make_feature_store_yaml(project, test_repo_config, repo_dir_name: Path):
         isinstance(config_dict["online_store"], dict)
         and "redis_type" in config_dict["online_store"]
     ):
-        config_dict["online_store"]["redis_type"] = "redis_cluster"
+        if str(config_dict["online_store"]["redis_type"]) == "RedisType.redis_cluster":
+            config_dict["online_store"]["redis_type"] = "redis_cluster"
+        elif str(config_dict["online_store"]["redis_type"]) == "RedisType.redis":
+            config_dict["online_store"]["redis_type"] = "redis"
     config_dict["repo_path"] = str(config_dict["repo_path"])
-
     return yaml.safe_dump(config_dict)