instance attribute support for functions

1 files changed, 2 insertions, 0 deletions

diff --git a/builder.py b/builder.py
index 7c7bca0..16b41be 100644
--- a/builder.py
+++ b/builder.py
@@ -175,6 +175,8 @@ class ASTNGBuilder:
                         # Const, Tuple, ... we may be wrong, may be not, but
                         # anyway we don't want to pollute builtin's namespace
                         continue
+                    elif infered.is_function:
+                        iattrs = infered.instance_attrs
                     else:
                         iattrs = infered.locals
                 except AttributeError: