Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion sdmetrics/visualization.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
"""Visualization methods for SDMetrics."""

import os
from functools import wraps

import pandas as pd
Expand Down Expand Up @@ -29,8 +30,11 @@ def wrapper(*args, **kwargs):
# Lazy import IPython
from IPython import get_ipython

env_plotly_renderer = os.environ.get('SDMETRICS_PLOTLY_RENDERER', None)
ipython_interpreter = str(get_ipython())
if 'ZMQInteractiveShell' in ipython_interpreter and 'iframe' in renderers:
if env_plotly_renderer is not None:
pio.renderers.default = env_plotly_renderer
elif 'ZMQInteractiveShell' in ipython_interpreter and 'iframe' in renderers:
# This means we are using jupyter notebook
pio.renderers.default = 'iframe'

Expand Down