Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
floriankrb committed Jul 15, 2024
1 parent a13be18 commit 65fec89
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 3 deletions.
5 changes: 2 additions & 3 deletions src/anemoi/registry/entry/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,13 +33,12 @@ class CatalogueEntry:

def __init__(self, key=None, path=None, must_exist=True):
assert key is not None or path is not None, "key or path must be provided"
assert key is None or path is None, "key and path are mutually exclusive"

if path is not None:
assert key is None
self.load_from_path(path)

if key is not None:
assert path is None
if self.key_exists(key):
# found in catalogue so load it
self.load_from_key(key)
Expand All @@ -48,7 +47,7 @@ def __init__(self, key=None, path=None, must_exist=True):
if must_exist:
raise CatalogueEntryNotFound(f"Could not find any {self.collection} with key={key}")
else:
self.create_from_new_new(key)
self.create_from_new_key(key)

assert self.record is not None
assert self.key is not None, "key must be provided"
Expand Down
7 changes: 7 additions & 0 deletions src/anemoi/registry/entry/experiment.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
import datetime
import logging
import os
from getpass import getuser

import yaml
from anemoi.utils.s3 import download
Expand All @@ -24,6 +25,12 @@ class ExperimentCatalogueEntry(CatalogueEntry):
collection = "experiments"
main_key = "expver"

def create_from_new_key(self, key):
assert self.key_exists(key) is False, f"{self.collection} with key={key} already exists"
metadata = dict(expver=key, user=getuser())
self.key = key
self.record = dict(expver=key, metadata=metadata, runs={})

def load_from_path(self, path):
assert os.path.exists(path), f"{path} does not exist"
assert path.endswith(".yaml"), f"{path} must be a yaml file"
Expand Down

0 comments on commit 65fec89

Please sign in to comment.