standardise aliases

src/sparsezoo/utils/registry.py

@@ -53,6 +53,17 @@ def standardize_lookup_name(name: str) -> str:
     return name.replace("_", "-").replace(" ", "-").lower()
 
 
+def standardize_alias_name(
+    name: Union[None, str, List[str]]
+) -> Union[None, str, List[str]]:
+    if name is None:
+        return None
+    elif isinstance(name, str):
+        return standardize_lookup_name(name)
+    else:  # isinstance(name, list)
+        return [standardize_lookup_name(n) for n in name]
+
+
 class RegistryMixin:
     """
     Universal registry to support registration and loading of child classes and plugins
@@ -210,6 +221,7 @@ def register(
         name = value.__name__
 
     name = standardize_lookup_name(name)
+    alias = standardize_alias_name(alias)
     register_alias(name=name, alias=alias, parent_class=parent_class)
 
     if require_subclass:

tests/sparsezoo/utils/test_registry.py

@@ -69,7 +69,7 @@ class Foo1(foo):
             pass
 
         assert {"foo1"} == set(foo.registered_names())
-        assert {"name-3", "name_4"} == set(foo.registered_aliases())
+        assert {"name-3", "name-4"} == set(foo.registered_aliases())
 
     def test_key_error_on_duplicate_alias(self, foo):
         # once we register an object under one alias, we can't
@@ -111,7 +111,7 @@ class Foo1(foo):
             pass
 
         assert {"name-2"} == set(foo.registered_names())
-        assert {"name-3", "name_4"} == set(foo.registered_aliases())
+        assert {"name-3", "name-4"} == set(foo.registered_aliases())
 
     def test_get_value_from_registry(self, foo):
         @foo.register(alias=["name-3"])