From d57f7e99552c29348867be7cf73da7e09ef7a6ad Mon Sep 17 00:00:00 2001 From: Martin Fitzner Date: Mon, 6 Jan 2025 11:49:04 +0100 Subject: [PATCH] Update type hints --- baybe/utils/validation.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/baybe/utils/validation.py b/baybe/utils/validation.py index 8c9037f6d..84b47ad2f 100644 --- 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: @@ -119,7 +119,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.