Change the unit test

Author: puzhichen

gpu4pyscf/grad/tests/test_tddft_opt.py

@@ -58,7 +58,7 @@ def test_opt_rhf_tda(self):
     def test_opt_rks_tda(self):
         mf = dft.RKS(mol, xc='b3lyp').to_gpu()
         mf.kernel()
-        td = mf.TDDFT().set(nstates=3)
+        td = mf.TDA().set(nstates=3)
         td.kernel()
         td_cpu = td.to_cpu()
         mol_gpu = optimize(td)