cosmetics

DagsHub · Jan 9, 2024 · 2e1e8ff · 2e1e8ff
1 parent dc4c0b0
commit 2e1e8ff
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dagshub/data_engine/client/data_client.py b/dagshub/data_engine/client/data_client.py
@@ -4,6 +4,7 @@
 import dacite
 import gql
 import rich.progress
+from gql.transport.exceptions import TransportQueryError
 from gql.transport.requests import RequestsHTTPTransport
 
 import dagshub.auth
@@ -161,9 +162,8 @@ def _exec(self, query: str, params: Optional[Dict[str, Any]] = None) -> Dict[str
         q = gql.gql(query)
         try:
             resp = self.client.execute(q, variable_values=params)
-        except gql.transport.exceptions.TransportQueryError as e:
-            print(f"Support-Id header: {self.client.transport.response_headers['X-DagsHub-Support-Id']}")
-            raise e
+        except TransportQueryError as e:
+            raise TransportQueryError(f"Support-Id: {self.client.transport.response_headers['X-DagsHub-Support-Id']}") from e
         return resp
 
     def _datasource_query(