Runs wind and PV forecasts for India and saves to database
Install dependencies (requires poetry)
poetry install
Lint with:
make lint
Format code with:
make format
make test
Replace {DB_URL}
with a postgres DB connection string (see below for setting up a ephemeral local DB)
If testing on a local DB, you may use the following script to seed the the DB with a dummy user, site and site_group.
DB_URL={DB_URL} poetry run seeder
This example runs the application and writes the results to stdout
DB_URL={DB_URL} NWP_ZARR_PATH={NWP_ZARR_PATH} poetry run app
To save batches, you need to set the SAVE_BATCHES_DIR
environment variable to directory.
### Starting a local database using docker
```bash
docker run \
-it --rm \
-e POSTGRES_USER=postgres \
-e POSTGRES_PASSWORD=postgres \
-p 54545:5432 postgres:14-alpine \
postgres
The corresponding DB_URL
will be
postgresql://postgres:postgres@localhost:54545/postgres
Building and running in Docker
Build the Docker image
make docker.build
Create a container from the image. This example runs the application and writes the results to stdout.
Replace {DB_URL}
with a postgres DB connection string.
N.B if the database host is localhost
on the host machine, replace localhost
with host.docker.internal
so that docker can access the database from within the container
docker run -it --rm -e DB_URL={DB_URL} -e NWP_ZARR_PATH={NWP_ZARR_PATH} ocf/india-forecast-app
This repo makes use of PyTorch (torch
and torchvision
packages) CPU-only version. In order to support installing PyTorch via poetry for various environments, we specify the exact wheels for each environment in the pyproject.toml file. Some background reading on why this is required can be found here: https://santiagovelez.substack.com/p/how-to-install-torch-cpu-in-poetry?utm_campaign=post&utm_medium=web&triedRedirect=true
Thanks goes to these wonderful people (emoji key):
priyanshubajaj |
Peter Dudfield 💻 |
Dakshbir 📖 |
MAYANK SHARMA 💻 |
This project follows the all-contributors specification. Contributions of any kind welcome!
Problem: poetry install
fails with dependency conflicts
Solution: Try updating Poetry first with pip install --upgrade poetry
, then run poetry update
followed by poetry install
Problem: Package installation errors
Solution: Check your Python version matches the one specified in pyproject.toml
. You can use poetry env use python3.x
to set the correct version.
Problem: Container can't connect to local database with "connection refused" error
Solution: If using localhost in your DB_URL, replace it with host.docker.internal
when running in Docker
Problem: Database authentication failures
Solution: Verify your DB_URL format is correct: postgresql://username:password@hostname:port/database
Problem: "Failed to load model" errors Solution: Ensure your HUGGINGFACE_TOKEN environment variable is set correctly. The token can be found in AWS Secret Manager under {environment}/huggingface/token.
Problem: Out of memory errors when loading models Solution: Ensure your system has sufficient RAM, or consider using a smaller model variant.