From 5aee3d046bd96d65f87660d7faa9310ef77b2b9e Mon Sep 17 00:00:00 2001 From: Kevin Su Date: Sun, 30 Jun 2024 11:41:29 +0800 Subject: [PATCH] refactor: Update task decorator to enable deck (#1701) --- examples/mlflow_plugin/README.md | 2 +- examples/nlp_processing/nlp_processing/word2vec_and_lda.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/mlflow_plugin/README.md b/examples/mlflow_plugin/README.md index 1510cc976..101bd6fab 100644 --- 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 ` 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 index a95846178..6e9eb0a0f 100644 --- 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(