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 @@ <h1>Source code for pytorch_tabnet.pretraining</h1><div class="highlight"><pre>
         <span class="n">weights</span><span class="p">,</span>
     <span class="p">):</span>
         <span class="bp">self</span><span class="o">.</span><span class="n">updated_weights</span> <span class="o">=</span> <span class="n">weights</span>
-        <span class="n">filter_weights</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">updated_weights</span><span class="p">)</span></div>
+        <span class="n">filter_weights</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">updated_weights</span><span class="p">)</span>
+        <span class="bp">self</span><span class="o">.</span><span class="n">preds_mapper</span> <span class="o">=</span> <span class="kc">None</span></div>
 
 <div class="viewcode-block" id="TabNetPretrainer.fit"><a class="viewcode-back" href="../../generated_docs/pytorch_tabnet.html#pytorch_tabnet.pretraining.TabNetPretrainer.fit">[docs]</a>    <span class="k">def</span> <span class="nf">fit</span><span class="p">(</span>
         <span class="bp">self</span><span class="p">,</span>
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"