|
import argparse |
|
import logging |
|
from auto_quantum_genius.config_loader import load_config |
|
from auto_quantum_genius.quantum_engine import QuantumEngine |
|
|
|
logger = logging.getLogger(__name__) |
|
|
|
def parse_args(): |
|
parser = argparse.ArgumentParser( |
|
description="Auto-Quantum-Genius: symbolic analysis with quantum heuristics" |
|
) |
|
parser.add_argument("--config", required=True, help="Path to YAML config file") |
|
parser.add_argument( |
|
"--mode", choices=["train", "infer"], default="infer", |
|
help="Operation mode" |
|
) |
|
return parser.parse_args() |
|
|
|
def main(): |
|
logging.basicConfig(level=logging.INFO) |
|
args = parse_args() |
|
cfg = load_config(args.config) |
|
engine = QuantumEngine(cfg) |
|
|
|
if args.mode == "train": |
|
result = engine.train() |
|
else: |
|
result = engine.infer() |
|
|
|
logger.info("Completed %s step; result summary: %s", args.mode, result) |
|
|
|
if __name__ == "__main__": |
|
main() |