1
0
mirror of https://github.com/TREX-CoE/trexio.git synced 2025-01-05 11:00:30 +01:00

Fix bug in python interface for writing CI coefficients

This commit is contained in:
Anthony Scemama 2023-06-02 15:09:02 +02:00
parent d8a37b26bf
commit cb629ccb03

View File

@ -4667,7 +4667,7 @@ def write_$group_dset$(trexio_file: File, offset_file: int, buffer_size: int, ds
if not isinstance(dset, (list, tuple, np.ndarray)): if not isinstance(dset, (list, tuple, np.ndarray)):
raise TypeError("dset argument has to be an array (list, tuple or NumPy ndarray).") raise TypeError("dset argument has to be an array (list, tuple or NumPy ndarray).")
if isinstance(dset, np.ndarray) and not coefficients.dtype==np.float64: if isinstance(dset, np.ndarray) and not dset.dtype==np.float64:
# convert to float64 if input is in a different precision # convert to float64 if input is in a different precision
dset_64 = np.float64(dset) dset_64 = np.float64(dset)
rc = pytr.trexio_write_safe_$group_dset$(trexio_file.pytrexio_s, offset_file, buffer_size, dset_64) rc = pytr.trexio_write_safe_$group_dset$(trexio_file.pytrexio_s, offset_file, buffer_size, dset_64)