From 5cfec11b37dc8d0d94f5b2efeaeba9b828343dc0 Mon Sep 17 00:00:00 2001 From: Brian McFee Date: Tue, 30 Jan 2024 13:22:12 -0500 Subject: [PATCH] connecting exception stack --- pescador/maps.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/pescador/maps.py b/pescador/maps.py index 0494c6c..cda1dbb 100644 --- a/pescador/maps.py +++ b/pescador/maps.py @@ -132,8 +132,8 @@ def buffer_stream(stream, buffer_size, partial=False, axis=None): continue try: yield __stack_data(data, axis=axis) - except (TypeError, AttributeError): - raise DataError(f"Malformed data stream: {data}") + except (TypeError, AttributeError) as ex: + raise DataError(f"Malformed data stream: {data}") from ex finally: data = [] count = 0 @@ -169,8 +169,8 @@ def tuples(stream, *keys): for data in stream: try: yield tuple(data[key] for key in keys) - except TypeError: - raise DataError(f"Malformed data stream: {data}") + except TypeError as ex: + raise DataError(f"Malformed data stream: {data}") from ex def keras_tuples(stream, inputs=None, outputs=None): @@ -230,8 +230,8 @@ def keras_tuples(stream, inputs=None, outputs=None): y = y[0] yield (x, y) - except TypeError: - raise DataError(f"Malformed data stream: {data}") + except TypeError as ex: + raise DataError(f"Malformed data stream: {data}") from ex def cache(stream, n_cache, prob=0.5, random_state=None):