diff --git a/analyze.py b/analyze.py index c0461bf2bf2482f0351bda751534e035c077b909_YW5hbHl6ZS5weQ==..9c06e009010b9983dcff3758c92b5cc01c80dc3a_YW5hbHl6ZS5weQ== 100644 --- a/analyze.py +++ b/analyze.py @@ -368,9 +368,9 @@ def visit(self, node, uid_func_mapping=None, kwargs=None, debug=False): # FIXME: not thread safe self.debug = debug - if uid_func_mapping: + if uid_func_mapping is not None: assert len(uid_func_mapping) <= 1 self.uid_func_mapping = uid_func_mapping self.uid_func = uid_func_mapping.values()[0] self.kwargs = kwargs self._visit(node) @@ -372,8 +372,11 @@ assert len(uid_func_mapping) <= 1 self.uid_func_mapping = uid_func_mapping self.uid_func = uid_func_mapping.values()[0] self.kwargs = kwargs self._visit(node) + if uid_func_mapping is not None: + self.uid_func_mapping = None + self.uid_func = None def visit_union(self, node): for select in node.children: