Python 2.7 compatibility

plotly · Jul 8, 2020 · 22087fb · 22087fb
1 parent f65482a
commit 22087fb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/repos/kaleido/py/kaleido/scopes/base.py b/repos/kaleido/py/kaleido/scopes/base.py
@@ -1,12 +1,16 @@
 import subprocess
 import json
 import base64
-from json import JSONDecodeError
 from threading import Lock, Thread
 import io
 import os
 import sys
 
+try:
+    from json import JSONDecodeError
+except ImportError:
+    JSONDecodeError = ValueError
+
 executable_path = os.path.join(
     os.path.dirname(os.path.dirname(os.path.abspath(__file__))),
     'executable',

diff --git a/repos/kaleido/py/kaleido/scopes/plotly.py b/repos/kaleido/py/kaleido/scopes/plotly.py
@@ -1,3 +1,4 @@
+from __future__ import absolute_import
 from kaleido.scopes.base import BaseScope
 from _plotly_utils.utils import PlotlyJSONEncoder
 import base64
@@ -108,7 +109,7 @@ def transform(self, figure, format=None, width=None, height=None, scale=None):
 
         # Base64 decode binary types
         if format not in self._text_formats:
-            img = base64.decodebytes(img)
+            img = base64.b64decode(img)
 
         return img