Remove hard dependency on TensorFlow (#237)

adriangb · Aug 2, 2021 · cd845db · cd845db
1 parent 7c072fb
commit cd845db
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/.github/workflows/tests.yaml b/.github/workflows/tests.yaml
@@ -76,7 +76,6 @@ jobs:
 
       - name: Install Dependencies
         run: |
-          poetry remove tensorflow
           poetry remove scikit-learn
           poetry install
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -33,9 +33,9 @@ version = "0.3.3"
 importlib-metadata = {version = "^3", python = "<3.8"}
 python = ">=3.6.1,<4.0"
 scikit-learn = ">=0.22.0"
-tensorflow = ">=2.4.0"
 
 [tool.poetry.dev-dependencies]
+tensorflow = ">=2.4.0"
 coverage = {extras = ["toml"], version = ">=5.4"}
 dataclasses = {version = "^0.8", python = "<3.7"}
 insipid-sphinx-theme = ">=0.2.2"

diff --git a/scikeras/__init__.py b/scikeras/__init__.py
@@ -9,6 +9,21 @@
 
 __version__ = importlib_metadata.version("scikeras")
 
+
+MIN_TF_VERSION = "2.4.0"
+TF_VERSION_ERR = f"SciKeras requires TensorFlow >= {MIN_TF_VERSION}."
+
+from packaging import version
+
+
+try:
+    from tensorflow import __version__ as tf_version
+except ImportError:  # pragma: no cover
+    raise ImportError("TensorFlow is not installed. " + TF_VERSION_ERR)
+else:
+    if version.parse(tf_version) < version.parse(MIN_TF_VERSION):  # pragma: no cover
+        raise ImportError(TF_VERSION_ERR)
+
 from tensorflow import keras as _keras
 
 from scikeras import _saving_utils