a yf@s2ddlZddlZddlZddlZddlmZddlZddlmZm Z m Z ddl m Z m Z ddlmZmZddlmZejjeddd  d d d d Zejjeddd  dd ddZejjddd ejjeddd  dd ddZejjeddd  dd ddZejjeddd  dd ddZdS)N)Path)MODELSOURCETMP)YOLOdownload) DATASETS_DIRSETTINGS)check_requirementsZrayF)installzray[tune] not installed)reasonc Cs"tdjdddddddddd S) z:Tune YOLO model using Ray for hyperparameter optimization.yolo11n-cls.yamlT imagenet10 Fcpu)Zuse_raydataZ grace_periodZ iterationsimgszepochsplotsdeviceN)rZtunerrS/var/www/html/django/DPS/env/lib/python3.9/site-packages/tests/test_integrations.pytest_model_ray_tunesrmlflowzmlflow not installedcCs,dtd<tdjdddddd dtd<d S) zQTest training with MLflow tracking enabled (see https://mlflow.org/ for details).Trr rrFrrrrrrN)r rtrainrrrr test_mlflowsrTzQTest failing in scheduled CI https://github.com/ultralytics/ultralytics/pull/8868cCsddl}dtd<d}|tjd<dtjd<td jd d d d dd|jj}|dks\Jd|jj }dtjd<td jd d d d dd|j |djj}|dksJdtj ddtd jd d d d dd|j |djj}|dksJdd td<dS)zVEnsure MLflow run status matches MLFLOW_KEEP_RUN_ACTIVE environment variable settings.rNTrzTest RunZ MLFLOW_RUNTrueZMLFLOW_KEEP_RUN_ACTIVEr rrrFrrRUNNINGzs(      7