From 2455f1dce37c3b7f8382ee57bf5e47ddb027cced Mon Sep 17 00:00:00 2001 From: Eduardo Carvalho Date: Tue, 2 Feb 2021 08:52:13 +0100 Subject: [PATCH] chore: release v3.1.1 --- CHANGELOG.md | 9 +++++++++ docs/_modules/pytorch_tabnet/pretraining.html | 3 ++- pyproject.toml | 2 +- 3 files changed, 12 insertions(+), 2 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index 08c0de67..0b76baf6 100755 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,4 +1,13 @@ +## [3.1.1](https://github.com/dreamquark-ai/tabnet/compare/v3.1.0...v3.1.1) (2021-02-02) + + +### Bug Fixes + +* add preds_mapper to pretraining ([76f2c85](https://github.com/dreamquark-ai/tabnet/commit/76f2c852f59c6ed2c5dc5f0766cb99310bae5f2c)) + + + # [3.1.0](https://github.com/dreamquark-ai/tabnet/compare/v3.0.0...v3.1.0) (2021-01-12) diff --git a/docs/_modules/pytorch_tabnet/pretraining.html b/docs/_modules/pytorch_tabnet/pretraining.html index c15c3eed..9244752b 100644 --- a/docs/_modules/pytorch_tabnet/pretraining.html +++ b/docs/_modules/pytorch_tabnet/pretraining.html @@ -197,7 +197,8 @@

Source code for pytorch_tabnet.pretraining

         weights,
     ):
         self.updated_weights = weights
-        filter_weights(self.updated_weights)
+ filter_weights(self.updated_weights) + self.preds_mapper = None
[docs] def fit( self, diff --git a/pyproject.toml b/pyproject.toml index b9950554..ac8f4b98 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "pytorch_tabnet" -version = "3.1.0" +version = "3.1.1" description = "PyTorch implementation of TabNet" homepage = "https://github.com/dreamquark-ai/tabnet" repository = "https://github.com/dreamquark-ai/tabnet"