add tensorflow to extra installs

README.rst

@@ -76,14 +76,11 @@ It's advised to install sklearn-genetic using a virtual env, inside the env use:
 
    pip install sklearn-genetic-opt
 
-If you want to get all the features, including plotting and mlflow logging capabilities,
+If you want to get all the features, including plotting, tensorboard and mlflow logging capabilities,
 install all the extra packages::
 
     pip install sklearn-genetic-opt[all]
 
-The only optional dependency that the last command does not install, it's Tensorflow,
-it is usually advised to look further which distribution works better for you.
-
 
 Example: Hyperparameters Tuning
 ###############################

docs/index.rst

@@ -53,8 +53,7 @@ and :class:`~sklearn_genetic.callbacks.TensorBoard` correspondingly.
 - MLflow (>= |MLflowMinVersion|)
 - Tensorflow (>= |TensorflowMinVersion|)
 
-This command will install all the extra requirements, except for Tensorflow,
-as it is usually advised to look further which distribution works better for you::
+This command will install all the extra requirements::
 
    pip install sklearn-genetic-opt[all]
 

setup.py

@@ -47,7 +47,8 @@
     extras_require={
         "mlflow": ["mlflow>=1.17.0"],
         "seaborn": ["seaborn>=0.11.2"],
-        "all": ["mlflow>=1.30.0", "seaborn>=0.11.2"],
+        "tensorflow": ["tensorflow>=2.0.0"],
+        "all": ["mlflow>=1.30.0", "seaborn>=0.11.2", "tensorflow>=2.0.0"],
     },
     python_requires=">=3.8",
     include_package_data=True,