fix[layoutmodel]: Fix type hints in LayoutPredictor::predict()

nikos-livathinos · nikos-livathinos · commit d6b8f0d26c3c · 2024-07-18T14:46:58.000+02:00
Signed-off-by: Nikos Livathinos &lt;nli@zurich.ibm.com&gt;
diff --git a/docling_ibm_models/layoutmodel/layout_predictor.py b/docling_ibm_models/layoutmodel/layout_predictor.py
@@ -118,7 +118,7 @@ def info(self) -> dict:
         }
         return info
 
-    def predict(self, orig_img: Union[Image, np.array]) -> Iterable[dict]:
+    def predict(self, orig_img: Union[Image.Image, np.ndarray]) -> Iterable[dict]:
         r"""
         Predict bounding boxes for a given image.
         The origin (0, 0) is the top-left corner and the predicted bbox coords are provided as:

Original file line number	Diff line number	Diff line change
`@@ -118,7 +118,7 @@ def info(self) -> dict:`
`118`	`118`	`}`
`119`	`119`	`return info`
`120`	`120`
`121`		`- def predict(self, orig_img: Union[Image, np.array]) -> Iterable[dict]:`
	`121`	`+ def predict(self, orig_img: Union[Image.Image, np.ndarray]) -> Iterable[dict]:`
`122`	`122`	`r"""`
`123`	`123`	`Predict bounding boxes for a given image.`
`124`	`124`	`The origin (0, 0) is the top-left corner and the predicted bbox coords are provided as:`