add axis to self.args_names

fepegar · Jul 6, 2024 · 08357f0 · 08357f0
1 parent c1cc856
commit 08357f0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/torchio/transforms/augmentation/intensity/random_motion.py b/src/torchio/transforms/augmentation/intensity/random_motion.py
@@ -166,6 +166,7 @@ def __init__(
  self.times = times
  self.image_interpolation = image_interpolation
  self.args_names = [
+ 'axis',
  'degrees',
  'translation',
  'times',