From 9a458b5a33351c93a8c0b9b2d1e76d2d2b8b14df Mon Sep 17 00:00:00 2001
From: "puzhichen.996" <puzhichen.996@bytedance.com>
Date: Mon, 20 Jan 2025 17:17:00 +0800
Subject: [PATCH] change some codes

---
 gpu4pyscf/grad/tdrhf.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/gpu4pyscf/grad/tdrhf.py b/gpu4pyscf/grad/tdrhf.py
index 20599aac..3c9d3f56 100644
--- a/gpu4pyscf/grad/tdrhf.py
+++ b/gpu4pyscf/grad/tdrhf.py
@@ -20,8 +20,8 @@
 from pyscf import lib
 from pyscf.lib import logger
 from gpu4pyscf.grad import rhf as rhf_grad
-from gpu4pyscf.df import int3c2e      #TODO: move int3c2e to out of df
-from gpu4pyscf.lib.cupy_helper import tag_array, contract, condense, sandwich_dot, reduce_to_device
+from gpu4pyscf.df import int3c2e     
+from gpu4pyscf.lib.cupy_helper import  contract
 from gpu4pyscf.scf import cphf
 from gpu4pyscf import lib as lib_gpu
 from pyscf import __config__