fix: update audit and propensity

rudderlabs · Dec 10, 2024 · 3e0278f · 3e0278f
1 parent 2a96a89
commit 3e0278f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/src/predictions/profiles_mlcorelib/py_native/id_stitcher/audit.py b/src/predictions/profiles_mlcorelib/py_native/id_stitcher/audit.py
@@ -26,8 +26,7 @@ class AuditIdStitcherModel(BaseModelType):
     }
 
     def __init__(self, build_spec: dict, schema_version: int, pb_version: str) -> None:
-        # FIXME: change output_type to "shell" once pb 0.20.0 is released
-        build_spec["materialization"] = {"output_type": "ephemeral"}
+        build_spec["materialization"] = {"output_type": "shell"}
         super().__init__(build_spec, schema_version, pb_version)
         self.recipe = ModelRecipe(self.build_spec)
 

diff --git a/src/predictions/profiles_mlcorelib/py_native/propensity.py b/src/predictions/profiles_mlcorelib/py_native/propensity.py
@@ -94,8 +94,7 @@ def __init__(
         parent_folder: WhtFolder,
         model_name: str,
     ) -> None:
-        # FIXME: change output_type to "none" once pb 0.20.0 is released
-        build_spec["materialization"] = {"output_type": "ephemeral"}
+        build_spec["materialization"] = {"output_type": "none"}
         super().__init__(build_spec, schema_version, pb_version)
         training_model_name = model_name + "_training"
         training_spec = self._get_training_spec()