Upgrade Python syntax

TorchIO-project · Jan 27, 2025 · 317fd09 · 317fd09
1 parent 3c1a7ff
commit 317fd09
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/torchio/transforms/augmentation/composition.py b/src/torchio/transforms/augmentation/composition.py
@@ -44,7 +44,7 @@ def __getitem__(self, index) -> Transform:
     def __repr__(self) -> str:
         return f'{self.name}({self.transforms})'
 
-    def get_base_args(self) -> Dict:
+    def get_base_args(self) -> dict:
         init_args = super().get_base_args()
         if 'parse_input' in init_args:
             init_args.pop('parse_input')
@@ -109,7 +109,7 @@ def __init__(self, transforms: TypeTransformsDict, **kwargs):
         super().__init__(parse_input=False, **kwargs)
         self.transforms_dict = self._get_transforms_dict(transforms)
 
-    def get_base_args(self) -> Dict:
+    def get_base_args(self) -> dict:
         init_args = super().get_base_args()
         if 'parse_input' in init_args:
             init_args.pop('parse_input')