github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

Move import location for Ray 2 (#721)

Browse Source 
* ray version check when importing

* display learner_class when starting_points removed

* test ray 2
This commit is contained in:
Chi Wang
2022-09-13 19:13:06 -07:00
committed by GitHub
parent a89efb552f
commit b7a010e657
17 changed files with 138 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
test/tune/example.py
View File

@@ -22,9 +22,15 @@ def easy_objective(config):
def test_blendsearch_tune(smoke_test=True):
try:
from ray import tune
from ray.tune.suggest import ConcurrencyLimiter
from ray.tune.schedulers import AsyncHyperBandScheduler
from ray.tune.suggest.flaml import BlendSearch
from ray import __version__ as ray_version
if ray_version.startswith("1."):
from ray.tune.suggest import ConcurrencyLimiter
from ray.tune.suggest.flaml import BlendSearch
else:
from ray.tune.search import ConcurrencyLimiter
from ray.tune.search.flaml import BlendSearch
except ImportError:
print("ray[tune] is not installed, skipping test")
return