1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
from colors import strip_color
class AwsError(Exception):
"""Raised when an AWS Exception is encountered."""
class FileExistsError(AwsError):
"""Raised when a File already exists on S3."""
class PapermillException(Exception):
"""Raised when an exception is encountered when operating on a notebook."""
class PapermillMissingParameterException(PapermillException):
"""Raised when a parameter without a value is required to operate on a notebook."""
class PapermillExecutionError(PapermillException):
"""Raised when an exception is encountered in a notebook."""
def __init__(self, cell_index, exec_count, source, ename, evalue, traceback):
args = cell_index, exec_count, source, ename, evalue, traceback
self.cell_index = cell_index
self.exec_count = exec_count
self.source = source
self.ename = ename
self.evalue = evalue
self.traceback = traceback
super().__init__(*args)
def __str__(self):
# Standard Behavior of an exception is to produce a string representation of its arguments
# when called with str(). In order to maintain compatibility with previous versions which
# passed only the message to the superclass constructor, __str__ method is implemented to
# provide the same result as was produced in the past.
message = f"\n{75 * '-'}\n"
message += f'Exception encountered at "In [{self.exec_count}]":\n'
message += strip_color("\n".join(self.traceback))
message += "\n"
return message
class PapermillRateLimitException(PapermillException):
"""Raised when an io request has been rate limited"""
class PapermillOptionalDependencyException(PapermillException):
"""Raised when an exception is encountered when an optional plugin is missing."""
class PapermillWarning(Warning):
"""Base warning for papermill."""
class PapermillParameterOverwriteWarning(PapermillWarning):
"""Callee overwrites caller argument to pass down the stream."""
def missing_dependency_generator(package, dep):
def missing_dep():
raise PapermillOptionalDependencyException(
f"The {package} optional dependency is missing. "
f"Please run pip install papermill[{dep}] to install this dependency"
)
return missing_dep
def missing_environment_variable_generator(package, env_key):
def missing_dep():
raise PapermillOptionalDependencyException(
f"The {package} optional dependency is present, but the environment "
f"variable {env_key} is not set. Please set this variable as "
f"required by {package} on your platform."
)
return missing_dep
|
