Allow unicode string type for py2

rbgirshick · Aug 20, 2018 · f5bb4f9 · f5bb4f9
1 parent f382098
commit f5bb4f9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -3,3 +3,4 @@ build
 yacs.egg-info
 yacs/__pycache__
 example/__pycache__
+.tox
diff --git a/yacs/config.py b/yacs/config.py
@@ -23,6 +23,11 @@
 
 # CfgNodes can only contain a limited set of valid types
 _VALID_TYPES = {dict, tuple, list, str, int, float, bool}
+# py2 allow for str and unicode
+try:
+    _VALID_TYPES = _VALID_TYPES.union({unicode})
+except Exception as e:
+    pass
 
 
 class CfgNode(dict):