suppress import errors for optional dependencies

microsoft · Jan 9, 2025 · f38c930 · f38c930
1 parent 38712b8
commit f38c930
Show file tree

Hide file tree

Showing 4 changed files with 288 additions and 251 deletions.
diff --git a/pe/__init__.py b/pe/__init__.py
@@ -1 +1,33 @@
-from .runner.pe import PE
+from generalimport import generalimport
+
+generalimport(
+    "blobfile",
+    "torch",
+    "imageio",
+    "clip",
+    "diffusers",
+    "wilds",
+    "improved_diffusion",
+    message=(
+        "Please install private_evolution with [image] dependencies. "
+        "See https://microsoft.github.io/DPSDA/getting_started/installation.html for more details."
+    ),
+)
+generalimport(
+    "gdown",
+    "openai",
+    "tenacity",
+    "azure-identity",
+    "tiktoken",
+    "python-dotenv",
+    "sentence-transformers",
+    "protobuf",
+    "sentencepiece",
+    "fschat",
+    "transformers",
+    "accelerate",
+    message=(
+        "Please install private_evolution with [text] dependencies. "
+        "See https://microsoft.github.io/DPSDA/getting_started/installation.html for more details."
+    ),
+)