o
    ci                     @   s   d dl mZ d dlmZ d dlmZmZ d dlmZ d dl	m
Z
 d dlmZ dd Zeeed	d
eedde
 gddZe Zedeejj dS )    )Path)train)	RunConfigScalingConfig)TorchTrainer)TBXLoggerCallback)MLflowLoggerCallbackc                  C   s"   t dD ]
} tt| d qd S )N   )epoch)ranger   reportdict)i r   \/home/ubuntu/.local/lib/python3.10/site-packages/ray/train/examples/mlflow_simple_example.py
train_func
   s   r      )num_workerstrain_experiment)experiment_name)	callbacks)scaling_config
run_configzRun directory:N)pathlibr   rayr   	ray.trainr   r   ray.train.torchr   ray.tune.loggerr   ray.tune.logger.mlflowr   r   trainerfitresultprintpathparentr   r   r   r   <module>   s$    