feat: conditionally import GitHub helper classes based on test data a…

…vailability
bhklab · Feb 4, 2025 · 5fd4cce · 5fd4cce
1 parent 23f39c1
commit 5fd4cce
Showing 1 changed file with 24 additions and 8 deletions.
diff --git a/src/imgtools/datasets/__init__.py b/src/imgtools/datasets/__init__.py
@@ -2,16 +2,32 @@
     data_images as example_data,
     data_paths as example_data_paths,
 )
-from .github_helper import (
-    GitHubRelease,
-    GitHubReleaseAsset,
-    MedImageTestData,
-)
 
 __all__ = [
-    "GitHubRelease",
-    "GitHubReleaseAsset",
-    "MedImageTestData",
     "example_data",
     "example_data_paths",
 ]
+
+
+def is_testdata_available() -> bool:
+    try:
+        from github import Github  # type: ignore # noqa
+        import aiohttp  # noqa
+
+        return True
+    except ImportError:
+        return False
+
+
+if is_testdata_available():
+    from .github_helper import (
+        GitHubRelease,
+        GitHubReleaseAsset,
+        MedImageTestData,
+    )
+
+    __all__ += [
+        "GitHubRelease",
+        "GitHubReleaseAsset",
+        "MedImageTestData",
+    ]