def transformed(self, X, energies, I_jumps):
try:
Km_Chi, Chi, bkgr = extra_exafs.extract_all(
energies, X, self.edge, I_jumps, self.extra_from,
self.extra_to, self.poly_deg, self.kweight, self.m)
except Exception as e:
# extra_exafs should be fixed to return a specific exception
if "jump at edge" in e.args[0]:
raise EdgeJumpException("Problem with edge jump.")
else:
raise
# this function always needs to return the expected input size - even
# if the test data was empty - force the output size
correct_shape = (X.shape[0], len(self.k_interp))
if Km_Chi.shape != correct_shape:
return np.full(correct_shape, np.nan)
return Km_Chi
def transformed(self, X, energies, I_jumps):
try:
Km_Chi, Chi, bkgr = extra_exafs.extract_all(energies, X,
self.edge, I_jumps,
self.extra_from, self.extra_to,
self.poly_deg, self.kweight, self.m)
except Exception as e:
# extra_exafs should be fixed to return a specific exception
if "jump at edge" in e.args[0]:
raise EdgeJumpException("Problem with edge jump.")
else:
raise
# this function always needs to return the expected input size - even
# if the test data was empty - force the output size
correct_shape = (X.shape[0], len(self.k_interp))
if Km_Chi.shape != correct_shape:
return np.full(correct_shape, np.nan)
return Km_Chi