Don't close MLflow run context when training model
Don't close MLflow run context when training model: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). This is the current version of train.py : python import mlflow from ludwig.api import LudwigModel from ludwig.contribs.mlflow import MlflowCallback import pandas as…
The task
This is the current version of `train.py`: ```python import mlflow from ludwig.api import LudwigModel from ludwig.contribs.mlflow import MlflowCallback import pandas as pd import pathlib import click
Part of FineEnvs/MiMo-V2.6-RL-harbor-code.