Skip to content

Commit b21f8fb

Browse files
committed
add psi4 to orbital ordering styles
1 parent 5e9e339 commit b21f8fb

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

qstack/reorder.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,10 @@ def __getitem__(self, l):
6868
'gpr': MagneticOrder(
6969
l_checker=lambda _: False,
7070
),
71+
'psi4': MagneticOrder(
72+
l_checker=lambda l: l>0,
73+
m_generator=_orca_get_m,
74+
),
7175
'orca': MagneticOrder(
7276
l_checker=lambda l: l>0,
7377
m_generator=_orca_get_m,
@@ -186,7 +190,7 @@ def reorder_ao(mol, vector, src='pyscf', dest='gpr'):
186190
If None, returns the indices to reorder and sign multipliers for an 1D vector
187191
to use as `x = x[idx]*sign`.
188192
src (str): Current convention. Defaults to 'pyscf'.
189-
dest (str): Convention to convert to (available: 'pyscf', 'gpr', 'orca', 'gaussian', 'turbomole'). Defaults to 'gpr'.
193+
dest (str): Convention to convert to (available: 'pyscf', 'gpr', 'orca', 'gaussian', 'turbomole', 'psi4'). Defaults to 'gpr'.
190194
191195
Returns:
192196
numpy.ndarray: Reordered vector or matrix, or tuple of (idx (numpy.ndarray), sign (numpy.ndarray)) if vector is None.

0 commit comments

Comments
 (0)