wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-10 12:30:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] Better handling of tune.function in global checkpoint (#4519)

Browse Source 
Enables result keys to be queried by CLI.
This commit is contained in:
Richard Liaw
2019-04-04 21:08:47 -07:00
committed by GitHub
parent fb88f7efe6
commit 50b2aa0740
3 changed files with 40 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/suggest/variant_generator.py
+14 -2
View File
@@ -58,7 +58,7 @@ class sample_from(object):
"""Specify that tune should sample configuration values from this function.
The use of function arguments in tune configs must be disambiguated by
either wrapped the function in tune.eval() or tune.function().
either wrapped the function in tune.sample_from() or tune.function().
Arguments:
func: An callable function to draw a sample from.
@@ -67,12 +67,18 @@ class sample_from(object):
def __init__(self, func):
self.func = func
def __str__(self):
return "tune.sample_from({})".format(str(self.func))
def __repr__(self):
return "tune.sample_from({})".format(repr(self.func))
class function(object):
"""Wraps `func` to make sure it is not expanded during resolution.
The use of function arguments in tune configs must be disambiguated by
either wrapped the function in tune.eval() or tune.function().
either wrapped the function in tune.sample_from() or tune.function().
Arguments:
func: A function literal.
@@ -84,6 +90,12 @@ class function(object):
def __call__(self, *args, **kwargs):
return self.func(*args, **kwargs)
def __str__(self):
return "tune.function({})".format(str(self.func))
def __repr__(self):
return "tune.function({})".format(repr(self.func))
_STANDARD_IMPORTS = {
"random": random,