[Model Monitoring] Add taosws.FetchError to caught exceptions

Addresses https://iguazio.atlassian.net/browse/ML-7806 following mlrun#6464 (comment)
jond01 · Oct 6, 2024 · 838231b · 838231b
1 parent 9708354
commit 838231b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: