style: add type hinting to distributions (queens-py#221)

* style: add type hinting to distributions

* style: include Copilot's review comments

* style: fix typos

* style: include Gil's feedback

Author: leahaeusel

.pre-commit-config.yaml

@@ -98,7 +98,7 @@ repos:
           (?x)(
           ^(.gitlab|.github|config|doc|tests|test_utils)/|
           ^src/(example_simulator_functions|queens_interfaces)/|
-          ^src/queens/(data_processors|distributions|drivers|iterators|models|parameters)/|
+          ^src/queens/(data_processors|drivers|iterators|models|parameters)/|
           ^src/queens/(schedulers|stochastic_optimizers|variational_distributions|visualization)/|
           ^src/queens/(main.py|global_settings.py)
           ).*$

pyproject.toml

@@ -164,11 +164,12 @@ disallow_untyped_calls = false  # Change to true once all functions are typed
 disallow_untyped_decorators = true
 disallow_untyped_defs = true
 ignore_missing_imports = true
+disable_error_code = ["str-bytes-safe"]
 # Delete modules once type hinting has been introduced there
 exclude = '''(?x)(
     ^(.gitlab|.github|config|doc|tests|test_utils)/|
     ^src/(example_simulator_functions|queens_interfaces)/|
-    ^src/queens/(data_processors|distributions|drivers|iterators|models|parameters)/|
+    ^src/queens/(data_processors|drivers|iterators|models|parameters)/|
     ^src/queens/(schedulers|stochastic_optimizers|variational_distributions|visualization)/|
     ^src/queens/(main.py|global_settings.py)
 ).*$'''

src/queens/distributions/__init__.py

@@ -21,6 +21,7 @@
 from queens.utils.imports import extract_type_checking_imports, import_class_from_class_module_map
 
 if TYPE_CHECKING:
+    from queens.distributions._distribution import Distribution
     from queens.distributions.bernoulli import Bernoulli
     from queens.distributions.beta import Beta
     from queens.distributions.categorical import Categorical
@@ -38,5 +39,5 @@
 class_module_map = extract_type_checking_imports(__file__)
 
 
-def __getattr__(name):
+def __getattr__(name: str) -> "Distribution":
     return import_class_from_class_module_map(name, class_module_map, __name__)