From 2bb3415aa0e0fd6b5066e14557aa5333948d72e7 Mon Sep 17 00:00:00 2001
From: Martin Lange <martin.lange@ufz.de>
Date: Tue, 29 Nov 2022 09:26:13 +0100
Subject: [PATCH] check for shared memory in subsequent CallbackOutput data
---
src/finam/sdk/output.py | 11 ++++++++++-
1 file changed, 10 insertions(+), 1 deletion(-)
diff --git a/src/finam/sdk/output.py b/src/finam/sdk/output.py
index ef5b3182..f921d944 100644
--- a/src/finam/sdk/output.py
+++ b/src/finam/sdk/output.py
@@ -393,6 +393,7 @@ class CallbackOutput(Output):
def __init__(self, callback, name, info=None, **info_kwargs):
super().__init__(name=name, info=info, static=False, **info_kwargs)
self.callback = callback
+ self.last_data = None
@property
def needs_push(self):
@@ -443,7 +444,15 @@ class CallbackOutput(Output):
raise FinamNoDataError(f"No data available in {self.name}")
with ErrorLogger(self.logger):
- return tools.to_xarray(data, self.name, self.info, time)
+ xdata = tools.to_xarray(data, self.name, self.info, time)
+ if self.last_data is not None and np.may_share_memory(
+ tools.get_magnitude(self.last_data), tools.get_magnitude(xdata)
+ ):
+ raise FinamDataError(
+ "Received data that shares memory with previously received data."
+ )
+ self.last_data = xdata
+ return xdata
def _check_time(time, is_static):
--
GitLab