update type for MolecularData ccsd amplitudes

qiskit-community · Jun 2, 2024 · 1e55524 · 1e55524
1 parent af595fe
commit 1e55524
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/ffsim/molecular_data.py b/python/ffsim/molecular_data.py
@@ -81,8 +81,8 @@ class MolecularData:
     mp2_t2: np.ndarray | None = None
     # CCSD data
     ccsd_energy: float | None = None
-    ccsd_t1: np.ndarray | None = None
-    ccsd_t2: np.ndarray | None = None
+    ccsd_t1: np.ndarray | tuple[np.ndarray, np.ndarray] | None = None
+    ccsd_t2: np.ndarray | tuple[np.ndarray, np.ndarray, np.ndarray] | None = None
     # FCI data
     fci_energy: float | None = None
     fci_vec: np.ndarray | None = None