diff --git a/obs/config.yaml b/dev/obs/config.yaml similarity index 97% rename from obs/config.yaml rename to dev/obs/config.yaml index 83755fc1..f2a2ffbf 100644 --- a/obs/config.yaml +++ b/dev/obs/config.yaml @@ -17,4 +17,4 @@ observations: # for now "observations" must be on top level, will be removed is_observations: true # will desappear - dataset: aifs-ea-an-oper-0001-mars-o96-1979-2022-6h-v6 start: 2004-11-09 00:00:00 - end: 2023-12-31 00:00:00 \ No newline at end of file + end: 2023-12-31 00:00:00 diff --git a/obs/demo.ipynb b/dev/obs/demo.ipynb similarity index 100% rename from obs/demo.ipynb rename to dev/obs/demo.ipynb diff --git a/obs/demo.py b/dev/obs/demo.py similarity index 80% rename from obs/demo.py rename to dev/obs/demo.py index 6ef01049..56b180b1 100755 --- a/obs/demo.py +++ b/dev/obs/demo.py @@ -1,10 +1,10 @@ #!/usr/bin/env python3 -import sys + from anemoi.datasets import open_dataset def str_(t): - """ Not needed, but useful for debugging """ + """Not needed, but useful for debugging""" import numpy as np if isinstance(t, (list, tuple)): @@ -15,6 +15,7 @@ def str_(t): return "{" + " , ".join(f"{k}: {str_(v)}" for k, v in t.items()) + "}" return str(t) + def main(path, filter): with open(path, "r") as f: import yaml @@ -43,11 +44,13 @@ def main(path, filter): data = ds[i] print(f"✅Got item {i} for time window ending {date}: {str_(data)}") -if __name__ == '__main__': + +if __name__ == "__main__": import argparse + parser = argparse.ArgumentParser() - parser.add_argument('--config', help='Path to config file', default='config.yaml') - parser.add_argument('--filter', help='filter dates (ex: 2017 or 2017-11)', default='2017-11-11') + parser.add_argument("--config", help="Path to config file", default="config.yaml") + parser.add_argument("--filter", help="filter dates (ex: 2017 or 2017-11)", default="2017-11-11") args = parser.parse_args() - main(args.config, args.filter) \ No newline at end of file + main(args.config, args.filter)