diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py index 53da9621..c9849ef9 100644 --- a/pvnet/models/base_model.py +++ b/pvnet/models/base_model.py @@ -282,10 +282,10 @@ def save_pretrained( minimize_data_config(new_data_config_path, new_data_config_path, self) # Get appropriate model card - model_name = huggingface_repo.split('/')[1] - if model_name == "windnet_india": + model_name = huggingface_repo.split("/")[1] + if model_name == "windnet_india": model_card = "windnet_model_card_template.md" - elif model_name == "pvnet_india": + elif model_name == "pvnet_india": model_card = "pvnet_india_model_card_template.md" else: model_card = "model_card_template.md" diff --git a/pvnet/models/model_cards/pvnet_india_model_card_template.md b/pvnet/models/model_cards/pvnet_india_model_card_template.md index 576538ec..36079953 100644 --- a/pvnet/models/model_cards/pvnet_india_model_card_template.md +++ b/pvnet/models/model_cards/pvnet_india_model_card_template.md @@ -12,7 +12,7 @@ ## Model Description -This model class uses numerical weather prediction and recent wind power output to forecast the PV power in North West India over the next 48 hours. More information can be found in the model repo [1] and experimental notes [here](https://github.com/openclimatefix/PVNet/tree/main/experiments/india). +This model class uses numerical weather prediction and recent wind power output to forecast the PV power in North West India over the next 48 hours. More information can be found in the model repo [1] and experimental notes [here](https://github.com/openclimatefix/PVNet/tree/main/experiments/india). - **Developed by:** openclimatefix diff --git a/pvnet/models/model_cards/windnet_model_card_template.md b/pvnet/models/model_cards/windnet_model_card_template.md index 8a8dfd85..189193e9 100644 --- a/pvnet/models/model_cards/windnet_model_card_template.md +++ b/pvnet/models/model_cards/windnet_model_card_template.md @@ -12,7 +12,7 @@ ## Model Description -This model class uses numerical weather prediction and recent wind power output to forecast the wind power in North West India over the next 48 hours at 15 minute granularity. More information can be found in the model repo [1] and experimental notes [here](https://github.com/openclimatefix/PVNet/tree/main/experiments/india). +This model class uses numerical weather prediction and recent wind power output to forecast the wind power in North West India over the next 48 hours at 15 minute granularity. More information can be found in the model repo [1] and experimental notes [here](https://github.com/openclimatefix/PVNet/tree/main/experiments/india). - **Developed by:** openclimatefix diff --git a/scripts/checkpoint_to_huggingface.py b/scripts/checkpoint_to_huggingface.py index 9fa36a9c..31903f90 100644 --- a/scripts/checkpoint_to_huggingface.py +++ b/scripts/checkpoint_to_huggingface.py @@ -18,7 +18,7 @@ def push_to_huggingface( checkpoint_dir_paths: list[str], - huggingface_repo: str = "openclimatefix/pvnet_uk_region", # or e.g. openclimatefix/windnet_india + huggingface_repo: str = "openclimatefix/pvnet_uk_region", # or e.g. openclimatefix/windnet_india wandb_repo: str | None = "openclimatefix/pvnet2.1", val_best: bool = True, wandb_ids: list[str] | None = [],