py: provide overloads for type-checking register_model

Signed-off-by: Isabella Basso do Amaral <[email protected]>

clients/python/src/model_registry/utils.py

@@ -4,9 +4,34 @@
 
 import os
 
+from typing_extensions import overload
+
 from .exceptions import StoreException
 
 
+@overload
+def s3_uri_from(
+    path: str,
+) -> str: ...
+
+
+@overload
+def s3_uri_from(
+    path: str,
+    bucket: str,
+) -> str: ...
+
+
+@overload
+def s3_uri_from(
+    path: str,
+    bucket: str,
+    *,
+    endpoint: str,
+    region: str,
+) -> str: ...
+
+
 def s3_uri_from(
     path: str,
     bucket: str | None = None,