[Typing][B-56] Add type annotations for python/paddle/utils/lazy_import.py (#65822)

Author: megemini

python/paddle/utils/lazy_import.py

@@ -13,12 +13,19 @@
 # limitations under the License.
 """Lazy imports for heavy dependencies."""
 
+from __future__ import annotations
+
 import importlib
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from types import ModuleType
+
 
 __all__ = []
 
 
-def try_import(module_name, err_msg=None):
+def try_import(module_name: str, err_msg: str | None = None) -> ModuleType:
     """Try importing a module, with an informative error message on failure."""
     install_name = module_name