Use the logging name instead of the one from fold_input

PiperOrigin-RevId: 870860151
Change-Id: I1d4485e1e08c3ca790899c33190b989d103eaf56
This commit is contained in:
Augustin Zidek
2026-02-16 06:13:13 -08:00
committed by Copybara-Service
parent e2b8ffd6a7
commit 2761bf1cfc

View File

@@ -447,9 +447,8 @@ class WholePdbPipeline:
and np.isnan(np.sum(value))
):
raise NanDataError(
'The output of the data pipeline contained nans. '
f'nan feature: {name}, fold input name: {fold_input.name}, '
f'random_seed {random_seed}'
f'Data pipeline output for {logging_name=} contains NaNs. NaN'
f' feature: {name}'
)
return np_example