Skip to content

Commit 9fe53e6

Browse files
committed
corrections as in #154
1 parent c0075e9 commit 9fe53e6

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

httomolibgpu/prep/phase.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
fftshift = Mock()
3939

4040
from numpy import float32
41-
from typing import Union
41+
from typing import Tuple
4242
import math
4343

4444
__all__ = [
@@ -377,7 +377,9 @@ def _calculate_pad_size(datashape: tuple) -> list:
377377
return pad_list
378378

379379

380-
def _pad_projections_to_second_power(tomo: cp.ndarray) -> Union[cp.ndarray, tuple]:
380+
def _pad_projections_to_second_power(
381+
tomo: cp.ndarray,
382+
) -> Tuple[cp.ndarray, Tuple[int, int]]:
381383
"""
382384
Performs padding of each projection to the next power of 2.
383385
If the shape is not even we also care of that before padding.

0 commit comments

Comments
 (0)