Refactor import paths for create_sample_dict in gates.py and mlflow.py to improve module organization