From 838231be496c4010702d8aa3eb0d4d9ec7b49961 Mon Sep 17 00:00:00 2001 From: Jonathan Daniel Date: Sun, 6 Oct 2024 20:30:50 +0200 Subject: [PATCH] [Model Monitoring] Add `taosws.FetchError` to caught exceptions Addresses https://iguazio.atlassian.net/browse/ML-7806 following https://github.com/mlrun/mlrun/pull/6464#discussion_r1789153446 --- mlrun/model_monitoring/db/tsdb/tdengine/tdengine_connector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mlrun/model_monitoring/db/tsdb/tdengine/tdengine_connector.py b/mlrun/model_monitoring/db/tsdb/tdengine/tdengine_connector.py index 136ac55210d8..2e1466bdc256 100644 --- a/mlrun/model_monitoring/db/tsdb/tdengine/tdengine_connector.py +++ b/mlrun/model_monitoring/db/tsdb/tdengine/tdengine_connector.py @@ -60,8 +60,8 @@ def connection(self) -> taosws.Connection: def with_retry_on_closed_connection(self, fn, **kwargs): try: return fn(self.connection, **kwargs) - except taosws.QueryError as err: - logger.warn(f"TDEngine query error: {err}") + except (taosws.QueryError, taosws.FetchError) as err: + logger.warn(f"TDEngine error: {err}") if "Internal error:" in str(err): logger.info("Retrying TDEngine query with a new connection") try: