ShaderCoder / utils /__init__.py
Vipitis's picture
improved docstring extraction
50c1955
from .tree_utils import (parse_functions, get_docstrings, grab_before_comments, line_chr2char, replace_function, get_root, node_str_idx, give_tree, full_func_head, has_docstrings)
from .html_utils import (make_iframe, make_script, construct_embed)
from .generation import (combine_generation_kwargs, stream_generation, construct_model_context)
tree_funcs = ["parse_functions", "get_docstrings", "grab_before_comments", "line_chr2char", "replace_function", "get_root", "node_str_idx", "give_tree", "full_func_head", "has_docstrings"]
html_funcs = ["make_iframe", "make_script", "construct_embed"]
gen_funcs = ["combine_generation_kwargs", "stream_generation", "construct_model_context"]
__all__ = tree_funcs + html_funcs + gen_funcs