fix: Simplify engine name

Signed-off-by: Harry <quanghai.ng1512@gmail.com>
feast-dev · Apr 9, 2024 · 11abe91 · 11abe91
1 parent 3e8a1f2
commit 11abe91
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/sdk/python/feast/repo_config.py b/sdk/python/feast/repo_config.py
@@ -47,7 +47,7 @@
     "snowflake.engine": "feast.infra.materialization.snowflake_engine.SnowflakeMaterializationEngine",
     "lambda": "feast.infra.materialization.aws_lambda.lambda_engine.LambdaMaterializationEngine",
     "bytewax": "feast.infra.materialization.contrib.bytewax.bytewax_materialization_engine.BytewaxMaterializationEngine",
-    "k8s.engine": "feast.infra.materialization.kubernetes.kubernetes_materialization_engine.KubernetesMaterializationEngine",
+    "k8s": "feast.infra.materialization.kubernetes.kubernetes_materialization_engine.KubernetesMaterializationEngine",
     "spark.engine": "feast.infra.materialization.contrib.spark.spark_materialization_engine.SparkMaterializationEngine",
 }
 

diff --git a/sdk/python/tests/integration/materialization/kubernetes/test_k8s.py b/sdk/python/tests/integration/materialization/kubernetes/test_k8s.py
@@ -24,7 +24,7 @@ def test_kubernetes_materialization():
         provider="aws",
         online_store={"type": "dynamodb", "region": "us-west-2"},
         offline_store_creator=RedshiftDataSourceCreator,
-        batch_engine={"type": "k8s.engine"},
+        batch_engine={"type": "k8s"},
         registry_location=RegistryLocation.S3,
     )
     env = construct_test_environment(config, None)