Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Implement get_next_trials #3107

Open
wants to merge 4 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
342 changes: 305 additions & 37 deletions ax/preview/api/client.py

Large diffs are not rendered by default.

12 changes: 4 additions & 8 deletions ax/preview/api/configs.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@

from dataclasses import dataclass, field
from enum import Enum
from typing import List, Optional
from typing import List, Mapping, Optional, Sequence

from ax.preview.api.types import TParameterValue

Expand Down Expand Up @@ -63,18 +63,14 @@ class ChoiceParameterConfig:
values: List[float] | List[int] | List[str] | List[bool]
parameter_type: ParameterType
is_ordered: bool | None = None
dependent_parameters: dict[TParameterValue, str] | None = None
dependent_parameters: Mapping[TParameterValue, Sequence[str]] | None = None


@dataclass
class ExperimentConfig:
"""
ExperimentConfig allows users to specify the SearchSpace and OptimizationConfig of
an Experiment and validates their inputs jointly.

This will also be the construct that handles transforming string-based inputs (the
objective, parameter constraints, and output constraints) into their corresponding
Ax class using SymPy.
ExperimentConfig allows users to specify the SearchSpace of an experiment along
with other metadata.
"""

name: str
Expand Down
Loading
Loading