when I run this code lines I get an error saying key error: \'input\' but I defined it in the train.yml file
import yaml data = yaml.safe_load(open(\'tra
