Adding current version print to version guard in __init__

tensorflow · Mar 8, 2016 · 3fc885f · 3fc885f
1 parent 21cf932
commit 3fc885f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/skflow/__init__.py b/skflow/__init__.py
@@ -29,6 +29,8 @@
 [pkg_rs.get_distribution(pkg).version for pkg in ['skflow', 'scikit-learn', 'tensorflow']]
 
 if SKLEARN_VERSION < '0.16.0':
-    raise ImportError("Your scikit-learn version needs to be at least 0.16. ")
+    raise ImportError("Your scikit-learn version needs to be at least 0.16. "
+                      "Your current version is %s. " % SKLEARN_VERSION)
 if TF_VERSION < '0.7.0':
-    raise ImportError("Your tensorflow version needs to be at least 0.7. ")
+    raise ImportError("Your tensorflow version needs to be at least 0.7. "
+                      "Your current version is %s. " % TF_VERSION)