upgraedd commited on
Commit
6661eeb
·
verified ·
1 Parent(s): e547165

Create main .py

Browse files
Files changed (1) hide show
  1. main .py +33 -0
main .py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import argparse
2
+ import logging
3
+ from auto_quantum_genius.config_loader import load_config
4
+ from auto_quantum_genius.quantum_engine import QuantumEngine
5
+
6
+ logger = logging.getLogger(__name__)
7
+
8
+ def parse_args():
9
+ parser = argparse.ArgumentParser(
10
+ description="Auto-Quantum-Genius: symbolic analysis with quantum heuristics"
11
+ )
12
+ parser.add_argument("--config", required=True, help="Path to YAML config file")
13
+ parser.add_argument(
14
+ "--mode", choices=["train", "infer"], default="infer",
15
+ help="Operation mode"
16
+ )
17
+ return parser.parse_args()
18
+
19
+ def main():
20
+ logging.basicConfig(level=logging.INFO)
21
+ args = parse_args()
22
+ cfg = load_config(args.config)
23
+ engine = QuantumEngine(cfg)
24
+
25
+ if args.mode == "train":
26
+ result = engine.train()
27
+ else:
28
+ result = engine.infer()
29
+
30
+ logger.info("Completed %s step; result summary: %s", args.mode, result)
31
+
32
+ if __name__ == "__main__":
33
+ main()