From b1ed0c7d22ba0d43118f7a12751d1809cf853e04 Mon Sep 17 00:00:00 2001
From: Dave Halter <davidhalter88@gmail.com>
Date: Wed, 27 Sep 2017 14:09:09 +0200
Subject: [PATCH] Add py__class__ to ContextSet.

---
 jedi/evaluate/context.py | 3 ++-
 jedi/evaluate/stdlib.py  | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/jedi/evaluate/context.py b/jedi/evaluate/context.py
index 6202e99f..98960c64 100644
--- 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()
diff --git a/jedi/evaluate/stdlib.py b/jedi/evaluate/stdlib.py
index 0f85b62d..447079c1 100644
--- a/jedi/evaluate/stdlib.py
+++ b/jedi/evaluate/stdlib.py
@@ -170,7 +170,7 @@ def builtins_type(evaluator, objects, bases, dicts):
         # It's a type creation... maybe someday...
         return NO_CONTEXTS
     else:
-        return ContextSet.from_iterable(o.py__class__() for o in objects)
+        return objects.py__class__()
 
 
 class SuperInstance(AbstractInstanceContext):