File size: 2,165 Bytes
9c6594c |
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 81 82 83 84 85 86 87 88 89 90 |
#
# Copyright (c) 2017-2021 NVIDIA CORPORATION. All rights reserved.
# This file is part of the WebDataset library.
# See the LICENSE file for licensing terms (BSD-style).
#
"""Pluggable exception handlers.
These are functions that take an exception as an argument and then return...
- the exception (in order to re-raise it)
- True (in order to continue and ignore the exception)
- False (in order to ignore the exception and stop processing)
They are used as handler= arguments in much of the library.
"""
import time
import warnings
def reraise_exception(exn):
"""Re-raise the given exception.
Args:
exn: The exception to be re-raised.
Raises:
The input exception.
"""
raise exn
def ignore_and_continue(exn):
"""Ignore the exception and continue processing.
Args:
exn: The exception to be ignored.
Returns:
bool: Always returns True to indicate continuation.
"""
return True
def warn_and_continue(exn):
"""Issue a warning for the exception and continue processing.
Args:
exn: The exception to be warned about.
Returns:
bool: Always returns True to indicate continuation.
"""
warnings.warn(repr(exn))
# Sleep ensures warnings don't scroll off screen too quickly
# These handlers are called rarely, so the sleep time isn't
# a significant performance bottleneck in practice
time.sleep(0.5)
return True
def ignore_and_stop(exn):
"""Ignore the exception and stop further processing.
Args:
exn: The exception to be ignored.
Returns:
bool: Always returns False to indicate stopping.
"""
return False
def warn_and_stop(exn):
"""Issue a warning for the exception and stop further processing.
Args:
exn: The exception to be warned about.
Returns:
bool: Always returns False to indicate stopping.
"""
warnings.warn(repr(exn))
# Sleep ensures warnings don't scroll off screen too quickly
# These handlers are called rarely, so the sleep time isn't
# a significant performance bottleneck in practice
time.sleep(0.5)
return False
|