module YAMLCompat

def self.load_file(path, permitted_classes: [])
  safe_load(File.read(path), permitted_classes: permitted_classes)
end

def self.safe_load(yaml_string, permitted_classes: [])
  if Psych::VERSION >= "4.0"
    YAML.safe_load(yaml_string, permitted_classes: permitted_classes)
  else
    YAML.safe_load(yaml_string, permitted_classes)
  end
end