Source code for ax.exceptions.data_provider
#!/usr/bin/env python3
# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
# pyre-strict
from collections.abc import Iterable
from typing import Any
[docs]
class DataProviderError(Exception):
"""Base Exception for Ax DataProviders.
The type of the data provider must be included.
The raw error is stored in the data_provider_error section,
and an Ax-friendly message is stored as the actual error message.
"""
def __init__(
self,
message: str,
data_provider: str,
# pyre-fixme[2]: Parameter annotation cannot be `Any`.
data_provider_error: Any,
) -> None:
self.message = message
self.data_provider = data_provider
self.data_provider_error = data_provider_error
def __str__(self) -> str:
return (
"{message}. \n Error thrown by: {dp} data provider \n"
+ "Native {dp} data provider error: {dp_error}"
).format(
dp=self.data_provider,
message=self.message,
dp_error=self.data_provider_error,
)
[docs]
class MissingDataError(Exception):
def __init__(self, missing_trial_indexes: Iterable[int]) -> None:
missing_trial_str = ", ".join([str(index) for index in missing_trial_indexes])
self.message: str = (
f"Unable to find data for the following trials: {missing_trial_str} "
"consider updating the data fetching kwargs or manually fetching "
"data via `refetch_data()`"
)
def __str__(self) -> str:
return self.message