File size: 609 Bytes
d4a00b2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#!/usr/bin/env python3
"""Entrypoint simples de inferencia para o modelo EDA treinado."""

import argparse

from train import analyze_schema


def main() -> int:
    parser = argparse.ArgumentParser(description="Gerar analise EDA com o modelo treinado.")
    parser.add_argument("--model-path", required=True, help="Caminho local ou repo Hub do modelo.")
    parser.add_argument("--input", required=True, help="Descricao do dataset a ser analisado.")
    args = parser.parse_args()

    print(analyze_schema(args.input, args.model_path))
    return 0


if __name__ == "__main__":
    raise SystemExit(main())