Fix Numpy 2.4 compatibility

PiperOrigin-RevId: 864751358
Change-Id: Ied340021bc1852a94fa8df1e176bddaff7902186
This commit is contained in:
Augustin Zidek
2026-02-03 01:26:16 -08:00
committed by Copybara-Service
parent 59cbb686da
commit dc61fb4a31

View File

@@ -460,7 +460,7 @@ def _merge_features_from_multiple_chains(
elif feature_name_split in TEMPLATE_FEATURES:
merged_example[feature_name] = np.concatenate(feats, axis=1)
elif feature_name_split in CHAIN_FEATURES:
merged_example[feature_name] = np.sum(x for x in feats).astype(np.int32)
merged_example[feature_name] = np.sum([x for x in feats]).astype(np.int32)
else:
merged_example[feature_name] = feats[0]
return merged_example