Allow custodian to work with generators.
Allow custodian to work with generators.

specter119/custodian,davidwaroquiers/custodian,materialsproject/custodian,materialsproject/custodian,specter119/custodian,xhqu1981/custodian,materialsproject/custodian,specter119/custodian