import os

def clean_imports(path):
    with open(path, 'r') as f:
        lines = f.readlines()
        
    bad_imports = [
        "DiarizationSegment",
        "SpeechSegment",
        "MusicSegment",
        "ImagingEquipment",
        "EADIdentifier",
        "LEIIdentifier",
        "AlternativeName",
        "Series",
        "MissionStatement",
        "StorageFacility",
        "AcquisitionBudget",
        "DigitizationBudget"
    ]
    
    new_lines = []
    for line in lines:
        stripped = line.strip()
        remove = False
        for bad in bad_imports:
            if bad in stripped and "modules/classes" in stripped:
                print(f"Removing {bad} from {path}")
                remove = True
                break
        
        if not remove:
            new_lines.append(line)
            
    with open(path, 'w') as f:
        f.writelines(new_lines)

clean_imports("schemas/20251121/linkml/01_custodian_name_modular.yaml")