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(