fix: optional dependency imports

Signed-off-by: Avik Basu <[email protected]>

numalogic/registry/__init__.py

@@ -13,17 +13,20 @@
 from numalogic.registry.artifact import ArtifactManager, ArtifactData, ArtifactCache
 from numalogic.registry.localcache import LocalLRUCache
 
+
+__all__ = ["ArtifactManager", "ArtifactData", "ArtifactCache", "LocalLRUCache"]
+
+
+try:
+ from numalogic.registry.mlflow_registry import MLflowRegistry # noqa: F401
+except ImportError:
+ pass
+else:
+ __all__.append("MLflowRegistry")
+
 try:
- from numalogic.registry.mlflow_registry import MLflowRegistry
- from numalogic.registry.redis_registry import RedisRegistry
+ from numalogic.registry.redis_registry import RedisRegistry # noqa: F401
 except ImportError:
- __all__ = ["ArtifactManager", "ArtifactData", "ArtifactCache", "LocalLRUCache"]
+ pass
 else:
- __all__ = [
- "ArtifactManager",
- "ArtifactData",
- "MLflowRegistry",
- "ArtifactCache",
- "LocalLRUCache",
- "RedisRegistry",
- ]
+ __all__.append("RedisRegistry")

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "numalogic"
-version = "0.4.dev4"
+version = "0.4.dev5"
 description = "Collection of operational Machine Learning models and tools."
 authors = ["Numalogic Developers"]
 packages = [{ include = "numalogic" }]
@@ -92,7 +92,7 @@ exclude = '''
 line-length = 100
 src = ["numalogic", "tests"]
 select = ["E", "F", "W", "C901", "NPY", "RUF", "TRY", "G", "PLE", "PLW", "UP", "ICN", "RET", "Q"]
-ignore = ["TRY003", "TRY301"]
+ignore = ["TRY003", "TRY301", "RUF100"]
 target-version = "py39"
 show-fixes = true
 show-source = true