Skip to content

Commit

Permalink
more typing for niftyreg
Browse files Browse the repository at this point in the history
Signed-off-by: neuronflow <[email protected]>
  • Loading branch information
neuronflow committed Dec 12, 2023
1 parent bd22647 commit 3f6df52
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions brainles_preprocessing/registration/niftyreg.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,9 +12,9 @@
class NiftyRegRegistrator(Registrator):
def __init__(
self,
registration_abspath=os.path.dirname(os.path.abspath(__file__)),
registration_script=None,
transformation_script=None,
registration_abspath: str = os.path.dirname(os.path.abspath(__file__)),
registration_script: str | None = None,
transformation_script: str | None = None,
):
"""
Initialize the NiftyRegRegistrator.
Expand Down Expand Up @@ -47,7 +47,7 @@ def register(
transformed_image_path: str,
matrix_path: str,
log_file_path: str,
):
) -> None:
"""
Register images using NiftyReg.
Expand Down Expand Up @@ -90,7 +90,7 @@ def transform(
transformed_image_path: str,
matrix_path: str,
log_file_path: str,
):
) -> None:
"""
Apply a transformation using NiftyReg.
Expand Down

0 comments on commit 3f6df52

Please sign in to comment.