Skip to content

Commit dfb90e1

Browse files
committed
BUG: Fill mask parameter for trivial model in model factory
Fill mask parameter for trivial model in model factory: extract the `mask` named argument from the keyword arguments. Fixes: ``` test/test_integration.py::test_proximity_estimator_trivial_model /home/runner/work/nifreeze/nifreeze/src/nifreeze/model/base.py:103: UserWarning: No mask provided; consider using a mask to avoid issues in model optimization. warn(mask_absence_warn_msg) ``` raised for example in: https://github.com/nipreps/nifreeze/actions/runs/12871532342/job/35885011589?pr=54#step:11:1159
1 parent 5025477 commit dfb90e1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/nifreeze/model/base.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,9 @@ def init(model="DTI", **kwargs):
5050
5151
"""
5252
if model.lower() in ("s0", "b0"):
53-
return TrivialModel(predicted=kwargs.pop("S0"), gtab=kwargs.pop("gtab"))
53+
return TrivialModel(
54+
mask=kwargs.pop("mask"), predicted=kwargs.pop("S0"), gtab=kwargs.pop("gtab")
55+
)
5456

5557
if model.lower() in ("avgdwi", "averagedwi", "meandwi"):
5658
from nifreeze.model.dmri import AverageDWIModel

0 commit comments

Comments
 (0)