Skip to content

Commit 7a0ba0d

Browse files
authored
[typing] fix type annotation error in DepthPro model image processor (#40238)
* fix type annotation error in DepthPro model image processor * fix * run make fix-copies
1 parent 00b4dfb commit 7a0ba0d

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/transformers/models/depth_pro/image_processing_depth_pro.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -318,7 +318,7 @@ def post_process_depth_estimation(
318318
self,
319319
outputs: "DepthProDepthEstimatorOutput",
320320
target_sizes: Optional[Union[TensorType, list[tuple[int, int]], None]] = None,
321-
) -> dict[str, list[TensorType]]:
321+
) -> list[dict[str, TensorType]]:
322322
"""
323323
Post-processes the raw depth predictions from the model to generate
324324
final depth predictions which is caliberated using the field of view if provided

src/transformers/models/depth_pro/image_processing_depth_pro_fast.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ def post_process_depth_estimation(
106106
self,
107107
outputs: "DepthProDepthEstimatorOutput",
108108
target_sizes: Optional[Union[TensorType, list[tuple[int, int]], None]] = None,
109-
) -> dict[str, list[TensorType]]:
109+
) -> list[dict[str, TensorType]]:
110110
"""
111111
Post-processes the raw depth predictions from the model to generate
112112
final depth predictions which is caliberated using the field of view if provided

0 commit comments

Comments
 (0)