Improve package checking to allow editable builds

Signed-off-by: Antoni Viros i Martin <[email protected]>

Author: ani300

fms_mo/utils/import_utils.py

@@ -17,11 +17,11 @@
 """
 
 # Standard
+from importlib.util import find_spec
 import pkgutil
 import sys
 
 # Third Party
-from transformers.utils.import_utils import _is_package_available
 import torch
 
 all_available_modules = []
@@ -47,7 +47,7 @@
 available_packages = {}
 for package in optional_packages:
     available_packages[package] = (
-        _is_package_available(package) or package in all_available_modules
+        find_spec(package) is not None or package in all_available_modules
     )
 
 # cutlass is detected through torch.ops.cutlass_gemm