Update type hints

emdgroup · Jan 6, 2025 · 0a42492 · 0a42492
1 parent 8213bef
commit 0a42492
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/baybe/utils/dataframe.py b/baybe/utils/dataframe.py
@@ -4,7 +4,7 @@
 
 import functools
 import logging
-from collections.abc import Callable, Collection, Iterable, Sequence
+from collections.abc import Callable, Iterable, Sequence
 from typing import TYPE_CHECKING, Literal, TypeVar, overload
 
 import numpy as np
@@ -70,7 +70,7 @@ def to_tensor(*x: np.ndarray | pd.DataFrame) -> Tensor | tuple[Tensor, ...]:
 
 def add_fake_measurements(
     data: pd.DataFrame,
-    targets: Collection[Target],
+    targets: Iterable[Target],
     good_reference_values: dict[str, list] | None = None,
     good_intervals: dict[str, tuple[float, float]] | None = None,
     bad_intervals: dict[str, tuple[float, float]] | None = None,
@@ -279,8 +279,8 @@ def add_parameter_noise(
 
 
 def create_fake_input(
-    parameters: Sequence[Parameter],
-    targets: Sequence[Target],
+    parameters: Iterable[Parameter],
+    targets: Iterable[Target],
     n_rows: int = 1,
     **kwargs: dict,
 ) -> pd.DataFrame:

diff --git a/baybe/utils/validation.py b/baybe/utils/validation.py
@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 import math
-from collections.abc import Callable, Sequence
+from collections.abc import Callable, Iterable
 from typing import TYPE_CHECKING, Any
 
 import pandas as pd
@@ -75,7 +75,7 @@ def validator(self: Any, attribute: Attribute, value: Any) -> None:
 """Validator for non-infinite floats."""
 
 
-def validate_target_input(data: pd.DataFrame, targets: Sequence[Target]) -> None:
+def validate_target_input(data: pd.DataFrame, targets: Iterable[Target]) -> None:
     """Validate input dataframe columns corresponding to targets.
 
     Args:
@@ -117,7 +117,7 @@ def validate_target_input(data: pd.DataFrame, targets: Sequence[Target]) -> None
 
 def validate_parameter_input(
     data: pd.DataFrame,
-    parameters: Sequence[Parameter],
+    parameters: Iterable[Parameter],
     numerical_measurements_must_be_within_tolerance: bool = False,
 ) -> None:
     """Validate input dataframe columns corresponding to parameters.