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

[Tune] Mlflow Integration (#12840)

Browse Source 
Co-authored-by: Kai Fricke <krfricke@users.noreply.github.com>
Co-authored-by: Richard Liaw <rliaw@berkeley.edu>
This commit is contained in:
Amog Kamsetty
2020-12-19 00:40:02 -08:00
committed by GitHub
parent 5d987f5988
commit 5d3c9c8861
18 changed files with 958 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/function_runner.py
+7 -2
View File
@@ -509,8 +509,9 @@ class FunctionRunner(Trainable):
try:
err_tb_str = self._error_queue.get(
block=block, timeout=ERROR_FETCH_TIMEOUT)
raise TuneError(("Trial raised an exception. Traceback:\n{}"
.format(err_tb_str)))
raise TuneError(
("Trial raised an exception. Traceback:\n{}".format(err_tb_str)
))
except queue.Empty:
pass
@@ -649,6 +650,10 @@ def with_parameters(fn, **kwargs):
def _inner(config):
inner(config, checkpoint_dir=None)
if hasattr(fn, "__mixins__"):
_inner.__mixins__ = fn.__mixins__
return _inner
if hasattr(fn, "__mixins__"):
inner.__mixins__ = fn.__mixins__
return inner