chore: Rename lambda to aws_lambda (feast-dev#3127)

* Add batch materialization engine docs to Gitbook structure Signed-off-by: Felix Wang <wangfelix98@gmail.com> * Add snowflake to `__init__.py` Signed-off-by: Felix Wang <wangfelix98@gmail.com> * Rename `lambda` to `aws_lambda` to allow Python imports Signed-off-by: Felix Wang <wangfelix98@gmail.com> * Format Signed-off-by: Felix Wang <wangfelix98@gmail.com> * Lint Signed-off-by: Felix Wang <wangfelix98@gmail.com> Signed-off-by: Felix Wang <wangfelix98@gmail.com>
srausser · Aug 24, 2022 · 66ee5d8 · 66ee5d8
1 parent 9af5bc2
commit 66ee5d8
Show file tree

Hide file tree

Showing 8 changed files with 11 additions and 2 deletions.
diff --git a/docs/SUMMARY.md b/docs/SUMMARY.md
@@ -99,6 +99,9 @@
   * [Google Cloud Platform](reference/providers/google-cloud-platform.md)
   * [Amazon Web Services](reference/providers/amazon-web-services.md)
   * [Azure](reference/providers/azure.md)
+* [Batch Materialization Engines](reference/batch-materialization/README.md)
+  * [Bytewax](reference/batch-materialization/bytewax.md)
+  * [Snowflake](reference/batch-materialization/snowflake.md)
 * [Feature repository](reference/feature-repository/README.md)
   * [feature\_store.yaml](reference/feature-repository/feature-store-yaml.md)
   * [.feastignore](reference/feature-repository/feast-ignore.md)

diff --git a/sdk/python/docs/index.rst b/sdk/python/docs/index.rst
@@ -340,7 +340,7 @@ Local Engine
 (Alpha) Lambda Based Engine
 ---------------------------
 
-.. automodule:: feast.infra.materialization.lambda.lambda_engine
+.. automodule:: feast.infra.materialization.aws_lambda.lambda_engine
     :members:
     :noindex:
 

diff --git a/sdk/python/feast/infra/materialization/__init__.py b/sdk/python/feast/infra/materialization/__init__.py
@@ -4,11 +4,17 @@
     MaterializationTask,
 )
 from .local_engine import LocalMaterializationEngine, LocalMaterializationJob
+from .snowflake_engine import (
+    SnowflakeMaterializationEngine,
+    SnowflakeMaterializationJob,
+)
 
 __all__ = [
     "MaterializationJob",
     "MaterializationTask",
     "BatchMaterializationEngine",
     "LocalMaterializationEngine",
     "LocalMaterializationJob",
+    "SnowflakeMaterializationEngine",
+    "SnowflakeMaterializationJob",
 ]
diff --git a/...t/infra/materialization/lambda/Dockerfile → ...fra/materialization/aws_lambda/Dockerfile b/...t/infra/materialization/lambda/Dockerfile → ...fra/materialization/aws_lambda/Dockerfile
diff --git a/.../infra/materialization/lambda/__init__.py → ...ra/materialization/aws_lambda/__init__.py b/.../infra/materialization/lambda/__init__.py → ...ra/materialization/aws_lambda/__init__.py
diff --git a/...feast/infra/materialization/lambda/app.py → ...t/infra/materialization/aws_lambda/app.py b/...feast/infra/materialization/lambda/app.py → ...t/infra/materialization/aws_lambda/app.py
diff --git a/...a/materialization/lambda/lambda_engine.py → ...terialization/aws_lambda/lambda_engine.py b/...a/materialization/lambda/lambda_engine.py → ...terialization/aws_lambda/lambda_engine.py
diff --git a/sdk/python/feast/repo_config.py b/sdk/python/feast/repo_config.py
@@ -37,7 +37,7 @@
 BATCH_ENGINE_CLASS_FOR_TYPE = {
     "local": "feast.infra.materialization.LocalMaterializationEngine",
     "snowflake.engine": "feast.infra.materialization.snowflake_engine.SnowflakeMaterializationEngine",
-    "lambda": "feast.infra.materialization.lambda.lambda_engine.LambdaMaterializationEngine",
+    "lambda": "feast.infra.materialization.aws_lambda.lambda_engine.LambdaMaterializationEngine",
     "bytewax": "feast.infra.materialization.contrib.bytewax.bytewax_materialization_engine.BytewaxMaterializationEngine",
 }