[tune] Support callable objects in variant generation (#6849)

* minorcallable

* format
This commit is contained in:
Richard Liaw
2020-01-21 10:24:25 -08:00
committed by GitHub
parent dac6268c5b
commit 4edfaf2f38
2 changed files with 25 additions and 2 deletions
+1 -2
View File
@@ -2,7 +2,6 @@ import copy
import logging
import numpy
import random
import types
from ray.tune import TuneError
from ray.tune.sample import sample_from
@@ -126,7 +125,7 @@ def _generate_variants(spec):
grid_vars = []
lambda_vars = []
for path, value in unresolved.items():
if isinstance(value, types.FunctionType):
if callable(value):
lambda_vars.append((path, value))
else:
grid_vars.append((path, value))