Skip to content

Commit 85f4fd8

Browse files
committed
psi4_FFI_to_strain.py: fix typehinting issue.
1 parent 6c20131 commit 85f4fd8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

psi4_FFI_to_strain.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -30,14 +30,14 @@
3030

3131
def read_psi4_dir(
3232
data_dir: str, ell_max: int, ell_min: int = 2
33-
) -> tuple[NDArray[np.float64], NDArray[np.complex128]]:
33+
) -> Tuple[NDArray[np.float64], NDArray[np.complex128]]:
3434
"""
3535
Read data from psi4 output directory and return time and mode data.
3636
3737
:param data_dir: The directory where psi4 output files are located.
3838
:param ell_max: Maximum spherical harmonic index to read.
3939
:param ell_min: Minimum spherical harmonic index to read, defaults to 2.
40-
:return: tuple[np.ndarray, np.ndarray]
40+
:return: Tuple[np.ndarray, np.ndarray]
4141
- time_data: Array of numpy.float64 time values (shape: (n_times,) ).
4242
- mode_data: 2D Array for modes of numpy.complex128 data (shape: (2*l+1, n_times,) ).
4343
:raises ValueError: Raises if the time arrays do not match in length for different l.
@@ -432,6 +432,7 @@ def quadratic(x: float, a: float, b: float, c: float) -> float:
432432
if results.failed > 0:
433433
print(f"Doctest failed: {results.failed} of {results.attempted} test(s)")
434434
sys.exit(1)
435+
print(f"Doctests passed: {results.attempted} of {results.attempted} test(s).")
435436

436437
if len(sys.argv) > 7:
437438
print("Error: Too many Arguments")

0 commit comments

Comments
 (0)