Source code for ax.exceptions.core
#!/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
[docs]
class AxError(Exception):
"""Base Ax exception.
All exceptions derived from AxError need to define a custom error message.
Additionally, exceptions can define a hint property that provides additional
guidance as to how to remedy the error.
"""
def __init__(self, message: str, hint: str = "") -> None:
self.message: str = message
self.hint: str = hint
def __str__(self) -> str:
return " ".join([self.message, getattr(self, "hint", "")]).rstrip()
[docs]
class UnsupportedError(AxError):
"""Raised when an unsupported request is made.
UnsupportedError may seem similar to NotImplementedError (NIE).
It differs in the following ways:
1. UnsupportedError is not used for abstract methods, which
is the official NIE use case.
2. UnsupportedError indicates an intentional and permanent lack of support.
It should not be used for TODO (another common use case of NIE).
"""
[docs]
class UnsupportedPlotError(AxError):
"""Raised when plotting functionality is not supported for the
given configurations.
"""
def __init__(self, message: str) -> None:
super().__init__(
message=message
or "Plotting functionality is not supported for the \
given configurations."
)
[docs]
class ExperimentNotReadyError(AxError):
"""Raised when failing to query data due to immature experiment.
Useful to distinguish data failure reasons in automated analyses.
"""
def __init__(
self, message: str, hint: str = "", exposures_unavailable: bool = False
) -> None:
super().__init__(message=message, hint=hint)
self.exposures_unavailable = exposures_unavailable
[docs]
class MetricDataNotReadyError(AxError):
"""Raised when trying to pull metric data from a trial that has
not finished running.
"""
pass
[docs]
class NoDataError(AxError):
"""Raised when no data is found for experiment in underlying data store.
Useful to distinguish data failure reasons in automated analyses.
"""
[docs]
class DataRequiredError(AxError):
"""Raised when more observed data is needed by the model to continue the
optimization.
Useful to distinguish when user needs to wait to request more trials until
more data is available.
"""
[docs]
class OptimizationComplete(AxError):
"""Raised when you hit SearchSpaceExhausted and GenerationStrategyComplete."""
def __init__(self, message: str) -> None:
super().__init__(
message=message
or "No more new points could be sampled, or maybe the underlying \
generation strategy has been completed."
)
[docs]
class OptimizationShouldStop(OptimizationComplete):
"""Raised when the Global Stopping Strategy suggests to stop the optimization."""
def __init__(self, message: str) -> None:
super().__init__(
message=message
or "The Global Stopping Strategy has decided to stop the optimization."
)
[docs]
class ObjectNotFoundError(AxError, ValueError):
"""Raised when an object is not found in the database.
This exception replaces ValueError raised by code when an objects is not
found in the database. In order to maintain backwards compatibility
ObjectNotFoundError inherits from ValueError. Dependency on ValueError
may be removed in the future.
"""
[docs]
class ExperimentNotFoundError(ObjectNotFoundError):
"""Raised when an experiment is not found in the database."""
[docs]
class SearchSpaceExhausted(OptimizationComplete):
"""Raised when using an algorithm that deduplicates points and no more
new points can be sampled from the search space."""
def __init__(self, message: str) -> None:
super().__init__(
message=message
or "No more new points could be sampled in the search space."
)
[docs]
class IncompatibleDependencyVersion(AxError):
"""Raise when an imcompatible dependency version is installed."""
[docs]
class AxWarning(Warning):
"""Base Ax warning.
All warnings derived from AxWarning need to define a custom warning message.
Additionally, warnings can define a hint property that provides additional
guidance as to how to remedy the warning.
"""
def __init__(self, message: str, hint: str = "") -> None:
self.message: str = message
self.hint: str = hint
def __str__(self) -> str:
return " ".join([self.message, getattr(self, "hint", "")]).rstrip()
[docs]
class AxStorageWarning(AxWarning):
"""Ax warning used for storage related concerns."""
[docs]
class AxParameterWarning(AxWarning):
"""Ax warning used for concerns related to parameter setups."""