aigrids 0.0.1

Datasets, Models and Methods for Power System Foundation Models

aidotgrids

Overview

1. Quick start
2. Datasets
3. Contributing

1. Quick start

Install stable version (option A):

pip install -U aidotgrids

Or, install bleeding-edge version (Option B):

pip uninstall -y aidotgrids         # ensure the PyPI version is removed
pip install git+https://github.com/AI-grids/aidotgrids

2. Datasets

All standardized tasks are hosted on Hugging Face Hub and can be downloaded automatically via aidotgrids.load. Current coverage:

Task / Dataset	Modality	Docs
OPFData	Graphs (optimal-power-flow)	details
PowerGraph	Transmission-grid graphs	details
SolarCube	Satellite imagery	details
BuildingElectricity	Time-series load profiles	details
WindFarm	SCADA & weather	details

Load a dataset in a few lines:

from aidotgrids import load

ds = load.load_task(
    task_name="WindFarm",
    subtask_name="odd_time_predict48h",
    root_path="~/AI-grids"  # local cache/download directory
)

3. Contributing

We welcome pull requests for new datasets, models and algorithms, or simply a fix in our code. Please read CONTRIBUTING.md before opening an issue or pull request.