Merge pull request deepchem#506 from lilleswing/lazy-import

Lazy Import xgboost
woodhaha · Apr 6, 2017 · 5ed5afa · 5ed5afa
2 parents 8a35de2 + 904b278
commit 5ed5afa
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/deepchem/models/xgboost_models/__init__.py b/deepchem/models/xgboost_models/__init__.py
@@ -2,7 +2,7 @@
 Scikit-learn wrapper interface of xgboost
 """
 
-import xgboost as xgb
+from deepchem.utils.dependencies import xgboost as xgb
 import numpy as np
 import os
 from deepchem.models import Model

diff --git a/deepchem/utils/dependencies.py b/deepchem/utils/dependencies.py
@@ -0,0 +1,11 @@
+"""
+This file is for lazily importing optional dependencies
+"""
+try:
+  import xgboost as xgboost
+except:
+  print("Warning: No XGBOOST installed on your system")
+  print(
+      "Attempting to run models with XGBOOST dependencies will throw runtime errors"
+  )
+  xgboost = None