Skip to content

Commit a062731

Browse files
committed
set inline matplotlib backend (#785)
1 parent 9c52c83 commit a062731

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

supervised/automl.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -436,6 +436,13 @@ def fit(
436436
raise e
437437
finally:
438438
matplotlib.use(original_backend)
439+
try:
440+
if 'inline' in original_backend:
441+
import matplotlib_inline
442+
matplotlib_inline.backend_inline._enable_matplotlib_integration()
443+
except:
444+
pass
445+
439446

440447
def predict(self, X: Union[List, numpy.ndarray, pandas.DataFrame]) -> numpy.ndarray:
441448
"""

0 commit comments

Comments
 (0)