Merge pull request #153 from bcaller/ast-cache

Cache of AST tree by module path

Author: KevinHock

pyt/core/ast_helper.py

@@ -4,6 +4,7 @@
 import ast
 import os
 import subprocess
+from functools import lru_cache
 
 
 BLACK_LISTED_CALL_NAMES = ['self']
@@ -21,6 +22,7 @@ def _convert_to_3(path):  # pragma: no cover
         exit(1)
 
 
+@lru_cache()
 def generate_ast(path):
     """Generate an Abstract Syntax Tree using the ast module.