set self type

holmesconan · Apr 12, 2016 · 62b7a00 · 62b7a00
1 parent 869736d
commit 62b7a00
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/main/java/org/yinwang/pysonar/visitor/TypeInferencer.java b/src/main/java/org/yinwang/pysonar/visitor/TypeInferencer.java
@@ -98,6 +98,7 @@ public Type visit(BinOp node, State s) {
             if (ltype instanceof InstanceType) {
                 Type opType = ltype.table.lookupAttrType("__sub__");
                     if (opType instanceof FunType) {
+                        ((FunType) opType).setSelfType(ltype);
                         return apply((FunType) opType, Collections.singletonList(rtype), null, null, null, node);
                     }
                 }