#!/usr/bin/env python3
import sys
from qpruner.utils import counter

# TODO: create parsers for training, inference etc
if __name__ == "__main__":
    if len(sys.argv) == 2:
        counter.count(sys.argv[1])
    else:
        counter.count(sys.argv[1], sys.argv[2])