Add masking_method arg name in Mask (#1137)

* Add masking_method as arguments in mask * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: Laurent Chauvin <[email protected]> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Fernando Pérez-García <[email protected]>
TorchIO-project · Jan 28, 2024 · b3c8f5b · b3c8f5b
1 parent 7464083
commit b3c8f5b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/torchio/transforms/preprocessing/intensity/mask.py b/src/torchio/transforms/preprocessing/intensity/mask.py
@@ -58,6 +58,7 @@ def __init__(
         self.masking_method = masking_method
         self.masking_labels = labels
         self.outside_value = outside_value
+        self.args_names = ['masking_method']
 
     def apply_transform(self, subject: Subject) -> Subject:
         for image in self.get_images(subject):