use doctest ignore import errors

Author: glemaitre

imblearn/keras/_generator.py

@@ -1,7 +1,7 @@
 """Implement generators for ``keras`` which will balance the data."""
 from __future__ import division
 
-import pytest
+import keras
 
 from scipy.sparse import issparse
 
@@ -10,8 +10,6 @@
 from sklearn.utils import check_random_state
 from sklearn.utils.testing import set_random_state
 
-keras = pytest.importorskip("keras")
-
 from ..under_sampling import RandomUnderSampler
 from ..utils import Substitution
 from ..utils._docstring import _random_state_docstring
@@ -141,7 +139,7 @@ def __getitem__(self, index):
                 self.indices_[index * self.batch_size:
                               (index + 1) * self.batch_size])
             if issparse(sample_weight_resampled) and not self.sparse:
-                sample_weight = sample_weight.toarray()
+                sample_weight_resampled = sample_weight_resampled.toarray()
 
         if self.sample_weight is None:
             return X_resampled, y_resampled

imblearn/tensorflow/_generator.py

@@ -2,17 +2,13 @@
 
 from __future__ import division
 
-import pytest
-
 from scipy.sparse import issparse
 
 from sklearn.base import clone
 from sklearn.utils import safe_indexing
 from sklearn.utils import check_random_state
 from sklearn.utils.testing import set_random_state
 
-tf = pytest.importorskip("tensorflow")
-
 from ..under_sampling import RandomUnderSampler
 from ..utils import Substitution
 from ..utils._docstring import _random_state_docstring

setup.cfg

@@ -35,4 +35,4 @@ doctest-fixtures = _fixture
 [tool:pytest]
 addopts = 
 	--doctest-modules
-
+        --doctest-ignore-import-errors