Skip to content

Commit

Permalink
STY: small fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
jwboth committed Jun 20, 2024
1 parent 244ad20 commit 676034e
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/darsia/restoration/averaging.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
"""Module with tools for volume averaging."""

from typing import Optional, Union, overload
from typing import Union, overload

import numpy as np
import scipy.ndimage
Expand Down Expand Up @@ -57,8 +57,10 @@ def __init__(self, rev: REV, mask: darsia.Image) -> None:

@overload # type: ignore [override]
def __call__(self, img: np.ndarray) -> np.ndarray: ...

@overload # type: ignore [override]
def __call__(self, img: darsia.Image) -> darsia.Image: ...

def __call__(
self, img: Union[np.ndarray, darsia.Image]
) -> Union[np.ndarray, darsia.Image]:
Expand Down

0 comments on commit 676034e

Please sign in to comment.