Skip to content

CLI

This module provides a command-line interface for generating power flow data.

main

Command-line interface for the data generation script.

Source code in gridfm_datakit/cli.py 
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
def main():
    """Command-line interface for the data generation script."""
    parser = argparse.ArgumentParser(
        description="Generate power flow data for grid analysis",
    )

    parser.add_argument(
        "config",
        type=str,
        help="Path to config file",
    )

    args = parser.parse_args()
    file_paths = generate_power_flow_data_distributed(args.config)

    print("\nData generation complete.")
    print("Generated files:")
    for key, path in file_paths.items():
        print(f"- {key}: {path}")