update registry

CHANGELOG.md

@@ -12,6 +12,7 @@ Keep it human-readable, your future self will thank you!
 
 ### Added
 - Add supporting_arrays to checkpoints
+- Add factories registry
 
 ## [0.4.1](https://github.com/ecmwf/anemoi-utils/compare/0.4.0...0.4.1) - 2024-10-23
 

src/anemoi/utils/registry.py

@@ -27,7 +27,12 @@ def __init__(self, package):
         self.registered = {}
         self.kind = package.split(".")[-1]
 
-    def register(self, name: str, factory: callable):
+    def register(self, name: str, factory: callable = None):
+
+        # Decorator version
+        if factory is None:
+            return lambda f: self.register(name, f)
+
         self.registered[name] = factory
 
     def _load(self, file):
@@ -77,3 +82,6 @@ def lookup(self, name: str) -> callable:
     def create(self, name: str, *args, **kwargs):
         factory = self.lookup(name)
         return factory(*args, **kwargs)
+
+    def __call__(self, name: str, *args, **kwargs):
+        return self.create(name, *args, **kwargs)