From 0e8c3c034623aff248dee939c4f1a61c4d162056 Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Fri, 11 Oct 2019 11:00:08 -0700 Subject: [PATCH] Don't wrap RayError with RayTaskError (#5870) --- python/ray/autoscaler/updater.py | 2 +- python/ray/exceptions.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/python/ray/autoscaler/updater.py b/python/ray/autoscaler/updater.py index 12397ebeb..e1547cf6f 100644 --- a/python/ray/autoscaler/updater.py +++ b/python/ray/autoscaler/updater.py @@ -293,7 +293,7 @@ class SSHCommandRunner(object): stdout=redirect, stderr=redirect) - def rsync_down(self, source, target, redirect=None): + def run_rsync_down(self, source, target, redirect=None): self.set_ssh_ip_if_required() self.process_runner.check_call( [ diff --git a/python/ray/exceptions.py b/python/ray/exceptions.py index 92897a3e7..3e6ea37f1 100644 --- a/python/ray/exceptions.py +++ b/python/ray/exceptions.py @@ -61,6 +61,9 @@ class RayTaskError(RayError): if issubclass(RayTaskError, self.cause_cls): return self # already satisfied + if issubclass(self.cause_cls, RayError): + return self # don't try to wrap ray internal errors + class cls(RayTaskError, self.cause_cls): def __init__(self, function_name, traceback_str, cause_cls, proctitle, pid, ip):