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: