ENH: fix deprecation warning by pkgutil.find_loader (#109)

xorbitsai · Nov 4, 2024 · 2945d9a · 2945d9a
1 parent 2e0d3f1
commit 2945d9a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/xoscar/utils.py b/python/xoscar/utils.py
@@ -19,11 +19,11 @@
 import dataclasses
 import functools
 import importlib
+import importlib.util as importlib_utils
 import inspect
 import io
 import logging
 import os
-import pkgutil
 import random
 import socket
 import sys
@@ -267,7 +267,7 @@ def add_load_handler(self, func: Callable):
             self._on_loads.append(func)
             return func
 
-    if pkgutil.find_loader(prefix_name) is not None:
+    if importlib_utils.find_spec(prefix_name) is not None:
         return LazyModule()
     elif placeholder:
         return ModulePlaceholder(prefix_name)