From c1e82139a724dc7b1793dd10ef693b1264d461cf Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 13 May 2019 11:41:49 +0200 Subject: [PATCH] Improved type hinting for image loading function Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index 2c8f94a..17c42e2 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -303,8 +303,8 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]], def _load_images_ssd_callback(resized_shape: Sequence[int]) \ - -> Callable[[Sequence[str]], - Tuple[tf.Tensor]]: + -> Callable[[Sequence[str], Sequence[Sequence[int]]], + Tuple[tf.Tensor, Sequence[Sequence[int]]]]: """ Returns the callback function to load images for SSD. @@ -315,7 +315,8 @@ def _load_images_ssd_callback(resized_shape: Sequence[int]) \ callback function """ - def _load_images_ssd(paths: Sequence[str], labels: Sequence[Sequence[int]]) -> Tuple[tf.Tensor]: + def _load_images_ssd(paths: Sequence[str], + labels: Sequence[Sequence[int]]) -> Tuple[tf.Tensor, Sequence[Sequence[int]]]: """ Callback function to load images for SSD.