fix: fix types

okunator · Jun 12, 2023 · c579722 · c579722
1 parent 7a5044d
commit c579722
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cellseg_models_pytorch/transforms/functional/generic_transforms.py b/cellseg_models_pytorch/transforms/functional/generic_transforms.py
@@ -288,7 +288,7 @@ def _check_operation_bounds(name: str, low: MAGNITUDE, high: MAGNITUDE) -> None:
 
 def _magnitude_kwargs(
     operation_name: str, bounds: Tuple[MAGNITUDE, MAGNITUDE], rng: RandomState
-) -> Optional[dict[str, MAGNITUDE]]:
+) -> Optional[Dict[str, MAGNITUDE]]:
     """Generate magnitude kwargs for apply_operations."""
     if operation_name == "tone":
         return dict(