From 1831a6a7c8c5330fe00520c5aa4ff957a025f365 Mon Sep 17 00:00:00 2001 From: Plamen Valentinov Kolev Date: Mon, 6 Nov 2023 16:16:21 +0100 Subject: [PATCH] Remove google colab --- sdmetrics/visualization.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/sdmetrics/visualization.py b/sdmetrics/visualization.py index b5d4b745..f7359595 100644 --- a/sdmetrics/visualization.py +++ b/sdmetrics/visualization.py @@ -27,14 +27,10 @@ def wrapper(*args, **kwargs): get_ipython = getattr(builtin, 'get_ipython') if hasattr(builtin, 'get_ipython') else False if get_ipython: ipython_interpreter = str(get_ipython()) - if 'colab' in ipython_interpreter and 'colab' in renderers: - pio.renderers.default = 'colab' - elif 'ZMQInteractiveShell' in ipython_interpreter and 'iframe' in renderers: + if 'ZMQInteractiveShell' in ipython_interpreter and 'iframe' in renderers: + # This means we are in jupyter notebook pio.renderers.default = 'iframe' - elif 'iframe' in renderers: - pio.renderers.default = 'iframe' - return function(*args, **kwargs) return wrapper