Remove google colab

sdv-dev · Nov 6, 2023 · 1831a6a · 1831a6a
1 parent d0d243f
commit 1831a6a
Showing 1 changed file with 2 additions and 6 deletions.
diff --git 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