import ignite and data separately (#256)

909935a5 · Gao, Xiang · GitHub · e30b98a8 · 909935a5
Unverified Commit 909935a5 authored Jun 19, 2019 by Gao, Xiang Committed by GitHub Jun 19, 2019
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

torchani/__init__.py torchani/__init__.py +4 -0

No files found.
--- a/torchani/__init__.py
+++ b/torchani/__init__.py
@@ -54,6 +54,10 @@ if sys.version_info[0] > 2:
    try:
        from . import ignite  # noqa: F401
        __all__.append('ignite')
+    except ImportError:
+        pass
+
+    try:
        from . import data  # noqa: F401
        __all__.append('data')
    except ImportError: