File size: 663 Bytes
590a501
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#!/usr/bin/env python3
"""Build qlib DatasetH / DataHandler artifacts from GP mining outputs."""

import argparse
import sys
from pathlib import Path

ROOT = Path(__file__).resolve().parents[1]
if str(ROOT) not in sys.path:
    sys.path.insert(0, str(ROOT))

from data_pipeline.gp_dataset_builder import export_gp_artifacts


def main():
    parser = argparse.ArgumentParser(description="Export GP features to qlib handler/dataset pickles")
    parser.add_argument("--run-id", type=str, default=None, help="GP mining run id (default: RUN_ID env)")
    args = parser.parse_args()
    export_gp_artifacts(run_id=args.run_id)


if __name__ == "__main__":
    main()