| class Action: | |
| equal = 0 | |
| replace = 1 | |
| insert = 2 | |
| delete = 3 | |
| Action.names = {i: name for name, i in Action.__dict__.items() if name[0] != "_"} | |
| import os | |
| from sourced.ml.transformers import Engine, UastExtractor, UastDeserializer, Transformer, HeadFiles | |
| from sourced.ml.utils import create_engine | |
| from sourced.ml.algorithms.uast_struct_to_bag import Uast2RandomWalks | |
| # Sourced-ml is based on transformers. | |
| # You create a pipeline from transformers and then run it. | |
| # We do not have a transformer, that just returns random walks, so we need to create it first. |
| uast { | |
| internal_type: "Module" | |
| children { | |
| internal_type: "FunctionDef" | |
| properties { | |
| key: "internalRole" | |
| value: "body" | |
| } | |
| properties { | |
| key: "returns" |