Merge pull request #70 from narumiruna/ruff

migrate to ruff

Makefile

@@ -2,7 +2,13 @@ install:
 	poetry install
 
 lint:
-	poetry run flake8 -v
+	poetry run ruff check .
 
 test:
 	poetry run pytest -v -s --cov=mlconfig tests
+
+publish:
+	poetry build -f wheel
+	poetry publish
+
+.PHONY: lint test publish

mlconfig/conf.py

@@ -3,8 +3,8 @@
 
 from omegaconf import OmegaConf
 
-_REGISTRY = {}
-_KEY_OF_FUNC_OR_CLS = "name"
+_registry = {}
+_key = "name"
 
 
 def load(f=None, obj=None) -> OmegaConf:
@@ -50,8 +50,8 @@ def _register(func_or_cls, name=None):
         if name is None:
             name = func_or_cls.__name__
 
-        if name not in _REGISTRY:
-            _REGISTRY[name] = func_or_cls
+        if name not in _registry:
+            _registry[name] = func_or_cls
         else:
             raise ValueError("duplicate name {} found".format(name))
 
@@ -64,14 +64,14 @@ def _register(func_or_cls, name=None):
 
 
 def getcls(conf):
-    return _REGISTRY[conf[_KEY_OF_FUNC_OR_CLS]]
+    return _registry[conf[_key]]
 
 
 def instantiate(conf, *args, **kwargs):
     kwargs = copy.deepcopy(kwargs)
 
     for k, v in conf.items():
-        if k not in kwargs and k != _KEY_OF_FUNC_OR_CLS:
+        if k not in kwargs and k != _key:
             kwargs[k] = v
 
     func_or_cls = getcls(conf)