1
0
Fork 0
mirror of https://github.com/fjosw/pyerrors.git synced 2025-03-16 15:20:24 +01:00

feat: hadrons npr input now returns rank-n tensors instead of 2D matrices

This commit is contained in:
Fabian Joswig 2021-12-01 14:28:21 +00:00
parent 4c8d758889
commit b715aa0c22

View file

@ -167,7 +167,7 @@ def read_ExternalLeg_hd5(path, filestem, ens_id, idl=None):
imag = Obs([rolled_array[si, sj, ci, cj].imag], [ens_id], idl=[idx]) imag = Obs([rolled_array[si, sj, ci, cj].imag], [ens_id], idl=[idx])
matrix[si, sj, ci, cj] = CObs(real, imag) matrix[si, sj, ci, cj] = CObs(real, imag)
return Npr_matrix(matrix.swapaxes(1, 2).reshape((12, 12), order='F'), mom_in=mom) return Npr_matrix(matrix, mom_in=mom)
def read_Bilinear_hd5(path, filestem, ens_id, idl=None): def read_Bilinear_hd5(path, filestem, ens_id, idl=None):
@ -219,7 +219,7 @@ def read_Bilinear_hd5(path, filestem, ens_id, idl=None):
imag = Obs([rolled_array[si, sj, ci, cj].imag], [ens_id], idl=[idx]) imag = Obs([rolled_array[si, sj, ci, cj].imag], [ens_id], idl=[idx])
matrix[si, sj, ci, cj] = CObs(real, imag) matrix[si, sj, ci, cj] = CObs(real, imag)
result_dict[key] = Npr_matrix(matrix.swapaxes(1, 2).reshape((12, 12), order='F'), mom_in=mom_in, mom_out=mom_out) result_dict[key] = Npr_matrix(matrix, mom_in=mom_in, mom_out=mom_out)
return result_dict return result_dict