Skip to content

Commit

Permalink
Update type hints
Browse files Browse the repository at this point in the history
  • Loading branch information
Scienfitz committed Jan 6, 2025
1 parent 25e717e commit d57f7e9
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions baybe/utils/validation.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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:
Expand Down Expand Up @@ -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.
Expand Down

0 comments on commit d57f7e9

Please sign in to comment.