def get_reference_files(datamodel): """Get the reference files associated with the step.""" refs = {} step = AssignWcsStep() for reftype in AssignWcsStep.reference_file_types: val = step.get_reference_file(datamodel, reftype) if val == 'N/A': refs[reftype] = None else: refs[reftype] = val return refs
def get_reference_files(datamodel): refs = {} step = AssignWcsStep() for reftype in AssignWcsStep.reference_file_types: val = step.get_reference_file(datamodel, reftype) print(reftype, val) if val.strip() == 'N/A': refs[reftype] = None else: refs[reftype] = val print(refs) return refs