Skip to content

Commit 7a36fbd

Browse files
authored
Add some missing args names (#1343)
1 parent 7b14a61 commit 7a36fbd

2 files changed

Lines changed: 9 additions & 0 deletions

File tree

src/torchio/transforms/preprocessing/spatial/crop_or_pad.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,14 @@ def __init__(
122122
raise ValueError(message)
123123
self.only_crop = only_crop
124124
self.only_pad = only_pad
125+
self.args_names = [
126+
'target_shape',
127+
'padding_mode',
128+
'mask_name',
129+
'labels',
130+
'only_crop',
131+
'only_pad',
132+
]
125133

126134
@staticmethod
127135
def _bbox_mask(mask_volume: np.ndarray) -> tuple[np.ndarray, np.ndarray]:

src/torchio/transforms/preprocessing/spatial/to_orientation.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@ def __init__(self, orientation: str = 'RAS', **kwargs):
6767
raise ValueError(message)
6868

6969
self.orientation = orientation
70+
self.args_names = ['orientation']
7071

7172
def apply_transform(self, subject: Subject) -> Subject:
7273
for image in subject.get_images(intensity_only=False):

0 commit comments

Comments
 (0)