https://gitlab.aicrowd.com/aicrowd/practice-challenges/aicrowd_OLNWP_challenge/raw/master/baselines/OLNWP_baseline.ipynb