chore: add exports for injector module

z3z1ma · Jul 17, 2024 · fd3383b · fd3383b
1 parent 65c1c54
commit fd3383b
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 3 deletions.
diff --git a/src/cdf/injector/__init__.py b/src/cdf/injector/__init__.py
@@ -1,4 +1,3 @@
-# flake8: noqa
 from cdf.injector.config import Config as Config
 from cdf.injector.config import get_config as get_config
 from cdf.injector.container import ConfigProxy as ConfigProxy
@@ -23,3 +22,30 @@
 from cdf.injector.specs import SingletonTuple as SingletonTuple
 from cdf.injector.specs import Spec as Spec
 from cdf.injector.specs import SpecID as SpecID
+
+__all__ = [
+    "Config",
+    "get_config",
+    "ConfigProxy",
+    "Container",
+    "get_container",
+    "ConfigError",
+    "FrozenConfigError",
+    "InputConfigError",
+    "NewKeyConfigError",
+    "PerturbSpecError",
+    "SetChildConfigError",
+    "Forward",
+    "GlobalInput",
+    "LocalInput",
+    "Object",
+    "Prototype",
+    "PrototypeMixin",
+    "Singleton",
+    "SingletonDict",
+    "SingletonList",
+    "SingletonMixin",
+    "SingletonTuple",
+    "Spec",
+    "SpecID",
+]
diff --git a/tests/injector/test_config.py b/tests/injector/test_config.py
@@ -1,4 +1,3 @@
-# mypy: disable-error-code="comparison-overlap"
 from __future__ import annotations
 
 import dataclasses

diff --git a/tests/injector/test_container.py b/tests/injector/test_container.py
@@ -1,4 +1,3 @@
-# mypy: disable-error-code="comparison-overlap"
 from __future__ import annotations
 
 from typing import Any, TypeVar, cast