tutela-app/scripts/diff2vec/make_sequences.py

import os
from typing import Any

from src.diff2vec.graph import UndirectedGraph
from src.diff2vec.euler import SubGraphSequences


def main(args: Any):
    graph: UndirectedGraph = UndirectedGraph()
    graph.from_pickle(args.edges_file)
    sequencer: SubGraphSequences = \
        SubGraphSequences(graph, args.cover_size, seed=args.seed)
    sequencer.get_sequences(args.components_file, args.sequences_file)


if __name__ == "__main__":
    from argparse import ArgumentParser
    parser: ArgumentParser = ArgumentParser()
    parser.add_argument('edges_file', type=str, help='path to edges pickle file')
    parser.add_argument('components_file', type=str, help='path to components file')
    parser.add_argument('sequences_file', type=str, help='path to sequences file')
    parser.add_argument('--cover-size', type=int, default=80, 
                        help='size of subgraph (default: 80)')
    parser.add_argument('--seed', type=int, default=42, help='random seed (default: 42)')
    args: Any = parser.parse_args()

    main(args)