Don't wrap RayError with RayTaskError (#5870)

This commit is contained in:
Eric Liang 2019-10-11 11:00:08 -07:00 committed by GitHub
parent 779f91523b
commit 0e8c3c0346
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 1 deletions

View file

@ -293,7 +293,7 @@ class SSHCommandRunner(object):
stdout=redirect, stdout=redirect,
stderr=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.set_ssh_ip_if_required()
self.process_runner.check_call( self.process_runner.check_call(
[ [

View file

@ -61,6 +61,9 @@ class RayTaskError(RayError):
if issubclass(RayTaskError, self.cause_cls): if issubclass(RayTaskError, self.cause_cls):
return self # already satisfied 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): class cls(RayTaskError, self.cause_cls):
def __init__(self, function_name, traceback_str, cause_cls, def __init__(self, function_name, traceback_str, cause_cls,
proctitle, pid, ip): proctitle, pid, ip):