refactor: Update task decorator to enable deck (#1701)

flyteorg · Jun 30, 2024 · 5aee3d0 · 5aee3d0
1 parent 4fbfa5d
commit 5aee3d0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/mlflow_plugin/README.md b/examples/mlflow_plugin/README.md
@@ -20,7 +20,7 @@ First, install the Flyte MLflow plugin:
 To log the metrics and parameters to Flyte deck, add {py:func}`@mlflow_autolog <flytekitplugins.mlflow.mlflow_autolog>` to the task. For example
 
 ```python
-@task(disable_deck=False)
+@task(enable_deck=True)
 @mlflow_autolog(framework=mlflow.keras)
 def train_model(epochs: int):
 ...

diff --git a/examples/nlp_processing/nlp_processing/word2vec_and_lda.py b/examples/nlp_processing/nlp_processing/word2vec_and_lda.py
@@ -285,7 +285,7 @@ def dimensionality_reduction(model_ser: FlyteFile[MODELSER_NLP]) -> plotdata:
     return x_vals, y_vals, labels
 
 
-@task(cache_version="1.0", cache=True, limits=Resources(mem="600Mi"), disable_deck=False)
+@task(cache_version="1.0", cache=True, limits=Resources(mem="600Mi"), enable_deck=True)
 def plot_with_plotly(x: List[float], y: List[float], labels: List[str]):
     layout = go.Layout(height=600, width=800)
     fig = go.Figure(