Skip to content

Commit cef9bfe

Browse files
committed
feat(alias): Add fill_value parameter to add_alias() - use in materilization
1 parent 5c4704f commit cef9bfe

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

UTILS/dfextensions/AliasDataFrame/AliasDataFrame.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4292,6 +4292,12 @@ def _do_materialize():
42924292
# Compute with context_override so dependent aliases can see prior results
42934293
result = self._eval_in_namespace(expr, context_override=results, alias_name=name)
42944294

4295+
# Apply fill_value for inf/NaN replacement (must be before dtype cast)
4296+
alias_spec = self._schema["columns"].get(name, {})
4297+
fill_val = alias_spec.get("fill_value")
4298+
if fill_val is not None:
4299+
result = np.where(np.isfinite(result), result, fill_val)
4300+
42954301
# Apply dtype if specified
42964302
result_dtype = self.alias_dtypes.get(name)
42974303
if result_dtype is not None:

0 commit comments

Comments
 (0)