--remote

pyflyte run

pyflyte run example_fyte.py training_workflow --hyperparameters '{"C": 0.1}'

FlyteScopedUserException: 'target'                                                                                                                                                                                                                                           
Failed with Unknown Exception <class 'KeyError'> Reason: 'Encountered error while executing workflow \'example.training_workflow\':\n  "Error encountered while executing \'training_workflow\':\\n  \'target\'"'
'Encountered error while executing workflow \'example.training_workflow\':\n  "Error encountered while executing \'training_workflow\':\\n  \'target\'"'

import pandas as pd
from sklearn.datasets import load_wine
from sklearn.linear_model import LogisticRegression

import flytekit.extras.sklearn
from flytekit import task, workflow


@task
def get_data() -> pd.DataFrame:
    """Get the wine dataset."""
    return load_wine(as_frame=True).frame

@task
def process_data(data: pd.DataFrame) -> pd.DataFrame:
    """Simplify the task from a 3-class to a binary classification problem."""
    return data.assign(target=lambda x: x["target"].where(x["target"] == 0, 1))

@task
def train_model(data: pd.DataFrame, hyperparameters: dict) -> LogisticRegression:
    """Train a model on the wine dataset."""
    features = data.drop("target", axis="columns")
    target = data["target"]
    return LogisticRegression(max_iter=3000, **hyperparameters).fit(features, target)