fixed lint issues and README

Author: RJ Agrawal

README.rst

@@ -382,13 +382,13 @@ NumericalTransformer takes function name as a string parameter and hence can be
     ... ])
     >>> mapper5.fit_transform(data)
     array([[1.38629436],
-               [1.79175947],
-               [1.09861229],
-               [1.09861229],
-               [0.69314718],
-               [1.09861229],
-               [1.60943791],
-               [1.38629436]])
+           [1.79175947],
+           [1.09861229],
+           [1.09861229],
+           [0.69314718],
+           [1.09861229],
+           [1.60943791],
+           [1.38629436]])
 
 
 

sklearn_pandas/transformers.py

@@ -1,7 +1,7 @@
 import numpy as np
 import pandas as pd
+from sklearn.base import TransformerMixin
 
-from sklearn.base import BaseEstimator, TransformerMixin
 
 def _get_mask(X, value):
     """
@@ -25,12 +25,13 @@ def __init__(self, func):
         """
         Params
 
-        func    function to apply to input columns. The function will be applied to each value.
-                Supported functions are defined in SUPPORTED_FUNCTIONS variable. Throws assertion
-                error if the not supported.
+        func    function to apply to input columns. The function will be
+                applied to each value. Supported functions are defined
+                in SUPPORTED_FUNCTIONS variable. Throws assertion error if the
+                not supported.
         """
         assert func in self.SUPPORTED_FUNCTIONS, \
-            f"Only following func arguments are supported: {self.SUPPORTED_FUNCTIONS}"
+            f"Only following func are supported: {self.SUPPORTED_FUNCTIONS}"
         super(NumericalTransformer, self).__init__()
         self.__func = func