From 77e49a904d1fd9b46442aafbaa312f63cf5aa015 Mon Sep 17 00:00:00 2001 From: "Kamil A. Kaczmarek" Date: Sat, 29 Sep 2018 20:40:07 +0200 Subject: [PATCH] Dev s12 (#38) * Update setup.py * Update models.py --- setup.py | 4 ++-- toolkit/lightgbm_transformers/models.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index e57e2bb..7da2228 100644 --- 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 index 4b5e400..6570440 100644 --- 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}