mirror of
https://github.com/microsoft/autogen.git
synced 2025-11-09 14:24:05 +00:00
ca9f9054e7
* categorical choice can be ordered or unordered * ordered -> order * move choice into utils * version comparison * packaging -> setuptools * import version * version_parse * test order for choice
29 lines
880 B
Python
29 lines
880 B
Python
from typing import Sequence
|
|
|
|
try:
|
|
from ray import __version__ as ray_version
|
|
|
|
assert ray_version >= "1.10.0"
|
|
from ray.tune import sample
|
|
except (ImportError, AssertionError):
|
|
from . import sample
|
|
|
|
|
|
def choice(categories: Sequence, order=None):
|
|
"""Sample a categorical value.
|
|
Sampling from ``tune.choice([1, 2])`` is equivalent to sampling from
|
|
``np.random.choice([1, 2])``
|
|
|
|
Args:
|
|
categories (Sequence): Sequence of categories to sample from.
|
|
order (bool): Whether the categories have an order. If None, will be decided autoamtically:
|
|
Numerical categories have an order, while string categories do not.
|
|
"""
|
|
domain = sample.Categorical(categories).uniform()
|
|
domain.ordered = (
|
|
order
|
|
if order is not None
|
|
else all(isinstance(x, (int, float)) for x in categories)
|
|
)
|
|
return domain
|