temporary fix to support fake import for azure.identity

microsoft · Jan 29, 2025 · 4305f65 · 4305f65
1 parent dbb45d6
commit 4305f65
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/pe/__init__.py b/pe/__init__.py
@@ -1,6 +1,10 @@
-from generalimport import generalimport
+import generalimport
+from generalimport import generalimport as optionalimport
 
-generalimport(
+# This is a temporary hack to support libraries such as azure.identity
+generalimport.top._assert_no_dots = lambda names: ...
+
+optionalimport(
     "blobfile",
     "torch",
     "imageio",
@@ -13,17 +17,17 @@
         "See https://microsoft.github.io/DPSDA/getting_started/installation.html for more details."
     ),
 )
-generalimport(
+optionalimport(
     "gdown",
     "openai",
     "tenacity",
-    "azure",
+    "azure.identity",
     "tiktoken",
     "dotenv",
     "sentence_transformers",
     "protobuf",
     "sentencepiece",
-    "fschat",
+    "fastchat",
     "transformers",
     "accelerate",
     message=(