Fixed a Python 2/3 compatibility issue

modelhub-ai · Dec 2, 2018 · 6cb52d2 · 6cb52d2
1 parent adbacfb
commit 6cb52d2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/framework/modelhubapi_tests/apitestbase.py b/framework/modelhubapi_tests/apitestbase.py
@@ -3,7 +3,11 @@
 import os
 from modelhubapi import ModelHubRESTAPI
 
-
+# Python 2/3 compatibility
+try:
+  basestring
+except NameError:
+  basestring = str
 
 class TestAPIBase(unittest.TestCase):
     """