Add py__class__ to ContextSet.

2017-09-27 14:09:09 +02:00
parent b74c8cb033
commit b1ed0c7d22
2 changed files with 3 additions and 2 deletions
--- a/jedi/evaluate/context.py
+++ b/jedi/evaluate/context.py
@@ -207,7 +207,8 @@ class ContextualizedName(ContextualizedNode):


 class ContextSet(BaseContextSet):
-    pass
+    def py__class__(self):
+        return ContextSet.from_iterable(c.py__class__() for c in self._set)


 NO_CONTEXTS = ContextSet()