Dev s12 (#38)

* Update setup.py * Update models.py
minerva-ml · Sep 29, 2018 · 77e49a9 · 77e49a9
1 parent e97b319
commit 77e49a9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -15,11 +15,11 @@
 
 setup(name='steppy-toolkit',
       packages=find_packages(),
-      version='0.1.11',
+      version='0.1.12',
       description='Set of tools to make your work with steppy faster and more effective.',
       long_description=long_description,
       url='https://github.com/minerva-ml/steppy-toolkit',
-      download_url='https://github.com/minerva-ml/steppy-toolkit/archive/0.1.11.tar.gz',
+      download_url='https://github.com/minerva-ml/steppy-toolkit/archive/0.1.12.tar.gz',
       author='Kamil A. Kaczmarek, Jakub Czakon',
       author_email='kamil.kaczmarek@neptune.ml, jakub.czakon@neptune.ml',
       keywords=['machine-learning', 'reproducibility', 'pipeline', 'tools'],

diff --git a/toolkit/lightgbm_transformers/models.py b/toolkit/lightgbm_transformers/models.py
@@ -69,7 +69,7 @@ def fit(self, X, y, X_valid, y_valid):
                                    **self.training_parameters)
         return self
 
-    def transform(self, X, y=None):
+    def transform(self, X, y=None, **kwargs):
         prediction = self.estimator.predict(X)
         return {'prediction': prediction}