from pathlib import Path
import json,sys
import yaml
ROOT=Path(__file__).resolve().parents[1];sys.path.insert(0,str(ROOT))
from model.fuxi_da import *
c=yaml.safe_load((ROOT/"conf/config.yaml").read_text()); sample=make_sample(c["data"]["tile_ids"][0],0,c["data"]["tile_size"],c["data"]["missing_probability"])
assert BACKGROUND_RAW_SHAPE==(70,721,1440) and sample["background"].shape==(70,c["data"]["tile_size"],c["data"]["tile_size"]) and sample["obs"].shape[0]==120
path=ROOT/c["data"]["manifest"];path.parent.mkdir(parents=True,exist_ok=True);path.write_text(json.dumps({"format_version":c["data"]["format_version"],"background_raw_shape":[70,721,1440],"background_model_shape":[70,720,1440],"agri_raw_shape":[8,15,640,640],"observation_model_shape":[120,640,640],"condition_shape":[190,640,640],"analysis_shape":[70,720,1440],"tile_ids":c["data"]["tile_ids"],"tile_size":c["data"]["tile_size"],"is_complete_global":False},indent=2));print(path)