def pe_for_tdscf(method, solvent_obj, dm=None): scf_solvent = getattr(method._scf, 'with_solvent', None) assert scf_solvent is None or isinstance(scf_solvent, PolEmbed) if not isinstance(solvent_obj, PolEmbed): solvent_obj = PolEmbed(method.mol, solvent_obj) return _attach_solvent._for_tdscf(method, solvent_obj, dm)
def ddpcm_for_tdscf(method, solvent_obj=None, dm=None): scf_solvent = getattr(method._scf, 'with_solvent', None) assert scf_solvent is None or isinstance(scf_solvent, DDPCM) if solvent_obj is None: solvent_obj = DDPCM(method.mol) return _attach_solvent._for_tdscf(method, solvent_obj, dm)