| import numpy as np | |
| from pathlib import Path | |
| def load_example_input(path: str): | |
| p = Path(path) | |
| if p.suffix == ".npz": | |
| z = np.load(p) | |
| return {k: z[k] for k in z.files} | |
| if p.suffix == ".npy": | |
| obj = np.load(p, allow_pickle=True) | |
| if isinstance(obj, np.ndarray) and obj.shape == (): | |
| return obj.item() | |
| if isinstance(obj, dict): | |
| return obj | |
| raise ValueError(f"Unsupported .npy content at {path}") | |
| raise ValueError(f"Unsupported input file type: {path}") | |
| def save_prediction_npz(path: str, u, v, w, k): | |
| Path(path).parent.mkdir(parents=True, exist_ok=True) | |
| np.savez(path, u=u, v=v, w=w, k=k) | |