Commit cffb63d3 authored by Stefan Behnel's avatar Stefan Behnel

Simplify code to format it better.

parent 640fe60b
......@@ -613,8 +613,8 @@ class TrackNumpyAttributes(VisitorTransform, SkipDeclarations):
def visit_AttributeNode(self, node):
self.visitchildren(node)
if (node.obj.is_name and node.obj.name in self.numpy_module_names) or \
node.obj.is_numpy_attribute:
obj = node.obj
if (obj.is_name and obj.name in self.numpy_module_names) or obj.is_numpy_attribute:
node.is_numpy_attribute = True
return node
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment