--- title: Title keywords: fastai sidebar: home_sidebar nb_path: "nbs/demo.ipynb" ---
! pip install -U tree-sitter==0.0.5
import function_parser
import pandas as pd
from function_parser.language_data import LANGUAGE_METADATA
from function_parser.process import DataProcessor
from tree_sitter import Language
language = 'python'
DataProcessor.PARSER.set_language(Language(function_parser.__path__[0] + '/tree-sitter-languages.so', language))
processor = DataProcessor(language=language,
language_parser=LANGUAGE_METADATA[language]['language_parser'])
dependee = 'keras-team/keras'
definitions = processor.process_dee(dependee, ext=LANGUAGE_METADATA[language]['ext'])
pd.DataFrame(definitions).head()
library_candidates = {}
library_candidates[dependee.split('/')[-1]] = definitions
dependent = dependee#'eriklindernoren/Keras-GAN'
calls, edges = processor.process_dent(dependent, ext=LANGUAGE_METADATA[language]['ext'], library_candidates=library_candidates)
edges[-5:]
pd.DataFrame(calls).head()
edges