Skip to content
This repository was archived by the owner on Sep 11, 2023. It is now read-only.

Commit 20db79f

Browse files
committed
update prepare_ml_data inputs
1 parent dad3d2f commit 20db79f

File tree

1 file changed

+6
-9
lines changed

1 file changed

+6
-9
lines changed

scripts/prepare_ml_data.py

+6-9
Original file line numberDiff line numberDiff line change
@@ -22,13 +22,10 @@
2222

2323
from nowcasting_dataset.dataset.datamodule import NowcastingDataModule
2424

25-
# from nowcasting_dataset.dataset.batch import write_batch_locally
2625
from nowcasting_dataset.data_sources.satellite.satellite_data_source import SAT_VARIABLE_NAMES
2726
from nowcasting_dataset.data_sources.nwp.nwp_data_source import NWP_VARIABLE_NAMES
28-
from nowcasting_dataset.dataset.batch import Batch
2927
from pathy import Pathy
3028
from pathlib import Path
31-
import fsspec
3229
import torch
3330
import os
3431
import numpy as np
@@ -55,20 +52,20 @@
5552
config = set_git_commit(config)
5653

5754
# Solar PV data
58-
PV_DATA_FILENAME = config.input_data.solar_pv_data_filename
59-
PV_METADATA_FILENAME = config.input_data.solar_pv_metadata_filename
55+
PV_DATA_FILENAME = config.input_data.pv.solar_pv_data_filename
56+
PV_METADATA_FILENAME = config.input_data.pv.solar_pv_metadata_filename
6057

6158
# Satellite data
62-
SAT_ZARR_PATH = config.input_data.satellite_zarr_path
59+
SAT_ZARR_PATH = config.input_data.satellite.satellite_zarr_path
6360

6461
# Numerical weather predictions
65-
NWP_ZARR_PATH = config.input_data.nwp_zarr_path
62+
NWP_ZARR_PATH = config.input_data.nwp.nwp_zarr_path
6663

6764
# GSP data
68-
GSP_ZARR_PATH = config.input_data.gsp_zarr_path
65+
GSP_ZARR_PATH = config.input_data.gsp.gsp_zarr_path
6966

7067
# Topographic data
71-
TOPO_TIFF_PATH = config.input_data.topographic_filename
68+
TOPO_TIFF_PATH = config.input_data.topographic.topographic_filename
7269

7370
# Paths for output data.
7471
DST_NETCDF4_PATH = Pathy(config.output_data.filepath)

0 commit comments

Comments
 (0)