Fix small errors according to lint

bytedance · Aug 30, 2024 · 2e91874 · 2e91874
1 parent 65f936d
commit 2e91874
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gpu4pyscf/scf/hf.py b/gpu4pyscf/scf/hf.py
@@ -17,6 +17,8 @@
 
 import copy
 import ctypes
+import sys
+import h5py
 import numpy as np
 import cupy
 import scipy.linalg
@@ -450,7 +452,7 @@ def _kernel(mf, conv_tol=1e-10, conv_tol_grad=None,
             local_variables = locals()
             for key in local_variables:
                 value = local_variables[key]
-                if (type(value) == cupy.ndarray):
+                if (type(value) is cupy.ndarray):
                     local_variables[key] = cupy.asnumpy(value)
             mf.dump_chk(local_variables)