Fixed issue with comet.ml in a notebook environment (2.1.2)

fbdesignpro · May 27, 2021 · 247d0c3 · 247d0c3
1 parent 06ea206
commit 247d0c3
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 4 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Change log
 
+#### 2.1.2 - 2021-05-27
+
+- **Fixed:** issue with comet.ml in a notebook environment
+
 #### 2.1.1 - 2021-05-27
 
 - **Fixed:** division by zero crash in some cases

diff --git a/README.md b/README.md
@@ -1,4 +1,4 @@
-![v](https://img.shields.io/badge/version-2.1.1-blue) ![v](https://img.shields.io/badge/updated-May%2027,%20%202021-green)
+![v](https://img.shields.io/badge/version-2.1.2-blue) ![v](https://img.shields.io/badge/updated-May%2027,%20%202021-green)
 
 ![Sweetviz Logo](http://cooltiming.com/SV/logo.png) 
 

diff --git a/sweetviz/__init__.py b/sweetviz/__init__.py
@@ -1,7 +1,7 @@
 # sweetviz public interface
 # -----------------------------------------------------------------------------------
 __title__ = 'sweetviz'
-__version__ = "2.1.1"
+__version__ = "2.1.2"
 __author__ = "Francois Bertrand"
 __license__ = 'MIT'
 

diff --git a/sweetviz/dataframe_report.py b/sweetviz/dataframe_report.py
@@ -40,7 +40,6 @@ def __init__(self,
         self._target = None
         self.test_mode = False
         self.corr_warning = list()
-        self._comet_ml_logger = comet_ml_logger.CometLogger()
         if fc is None:
             fc = FeatureConfig()
 
@@ -544,6 +543,7 @@ def show_html(self, filepath='SWEETVIZ_REPORT.html', open_browser=True, layout='
                   "Affected correlations:" + str(self.corr_warning))
 
         # Auto-log to comet_ml if desired & present
+        self._comet_ml_logger = comet_ml_logger.CometLogger()
         if self._comet_ml_logger._logging:
             self.generate_comet_friendly_html()
             self._comet_ml_logger.log_html(self._page_html)
@@ -592,7 +592,8 @@ def show_notebook(self, w=None, h=None, scale=None, layout=None, filepath=None):
                   "(likely due to only a single row containing non-NaN values for both correlated features)\n"
                   "Affected correlations:" + str(self.corr_warning))
 
-       # Auto-log to comet_ml if desired & present
+        # Auto-log to comet_ml if desired & present
+        self._comet_ml_logger = comet_ml_logger.CometLogger()
         if self._comet_ml_logger._logging:
             self.generate_comet_friendly_html()
             self._comet_ml_logger.log_html(self._page_html)