mirror of
https://github.com/microsoft/autogen.git
synced 2025-10-27 15:59:35 +00:00
b7a010e657
* ray version check when importing * display learner_class when starting_points removed * test ray 2
41 lines
956 B
Python
41 lines
956 B
Python
try:
|
|
from ray import __version__ as ray_version
|
|
|
|
assert ray_version >= "1.10.0"
|
|
from ray.tune import (
|
|
uniform,
|
|
quniform,
|
|
randint,
|
|
qrandint,
|
|
randn,
|
|
qrandn,
|
|
loguniform,
|
|
qloguniform,
|
|
lograndint,
|
|
qlograndint,
|
|
)
|
|
|
|
if ray_version.startswith("1."):
|
|
from ray.tune import sample
|
|
else:
|
|
from ray.tune.search import sample
|
|
except (ImportError, AssertionError):
|
|
from .sample import (
|
|
uniform,
|
|
quniform,
|
|
randint,
|
|
qrandint,
|
|
randn,
|
|
qrandn,
|
|
loguniform,
|
|
qloguniform,
|
|
lograndint,
|
|
qlograndint,
|
|
)
|
|
from . import sample
|
|
from .tune import run, report, INCUMBENT_RESULT
|
|
from .sample import polynomial_expansion_set
|
|
from .sample import PolynomialExpansionSet, Categorical, Float
|
|
from .trial import Trial
|
|
from .utils import choice
|