Fix a crash caused by accessing is_enum on a non-class.

Noticed this while testing another change. PiperOrigin-RevId: 482555441
google · Oct 20, 2022 · e53ce6c · e53ce6c
1 parent a425494
commit e53ce6c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pytype/vm.py b/pytype/vm.py
@@ -103,7 +103,9 @@ def add_branch(self, op: opcodes.Opcode, match_var: cfg.Variable,
     except abstract_utils.ConversionError:
       return None
     # We currently only implement exhaustiveness matching for enums.
-    if not (isinstance(match_val, abstract.Instance) and match_val.cls.is_enum):
+    if not (isinstance(match_val, abstract.Instance) and
+            isinstance(match_val.cls, abstract.Class) and
+            match_val.cls.is_enum):
       return None
     if not (isinstance(case_val, abstract.Instance) and
             case_val.cls == match_val.cls):