Fixed memory problem when logging multiple importance charts (#80)

* Fixed memory problem when logging multiple importance charts * version update
neptune-ai · Apr 12, 2020 · f027343 · f027343
1 parent 262057b
commit f027343
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -52,7 +52,7 @@
 # The short X.Y version
 version = '0.18'
 # The full version, including alpha/beta/rc tags
-release = '0.18.2'
+release = '0.18.3'
 
 # -- General configuration ---------------------------------------------------
 

diff --git a/neptunecontrib/monitoring/xgboost_monitor.py b/neptunecontrib/monitoring/xgboost_monitor.py
@@ -218,8 +218,13 @@ def _log_model(booster, name, npt):
 
 
 def _log_importance(booster, max_num_features, npt, **kwargs):
+    try:
+        import matplotlib.pyplot as plt
+    except ImportError:
+        raise ImportError('Please install matplotlib to log importance')
     importance = xgb.plot_importance(booster, max_num_features=max_num_features, **kwargs)
     npt.log_image('feature_importance', importance.figure)
+    plt.close('all')
 
 
 def _log_trees(booster, tree_list, img_name, npt, **kwargs):

diff --git a/setup.py b/setup.py
@@ -23,7 +23,7 @@ def main():
 
     setup(
         name='neptune-contrib',
-        version='0.18.2',
+        version='0.18.3',
         description='Neptune.ai contributions library',
         author='neptune.ai',
         support='[email protected]',