Changed error type

jprakash-db · jprakash-db · commit 1f3bc1937bfe · 2024-12-13T00:03:08.000+05:30
diff --git a/src/databricks/sql/thrift_backend.py b/src/databricks/sql/thrift_backend.py
@@ -1046,8 +1046,8 @@ def fetch_results(
 
         resp = self.make_request(self._client.FetchResults, req)
         if resp.results.startRowOffset > expected_row_start_offset:
-            raise ValueError(
-                "Expected results to start from {} but they instead start at {}".format(
+            raise DataError(
+                "fetch_results failed due to inconsistency in the state between the client and the server. Expected results to start from {} but they instead start at {}, some result batch must have been skipped".format(
                     expected_row_start_offset, resp.results.startRowOffset
                 )
             )

Original file line number	Diff line number	Diff line change
`@@ -1046,8 +1046,8 @@ def fetch_results(`
`1046`	`1046`
`1047`	`1047`	`resp = self.make_request(self._client.FetchResults, req)`
`1048`	`1048`	`if resp.results.startRowOffset > expected_row_start_offset:`
`1049`		`- raise ValueError(`
`1050`		`- "Expected results to start from {} but they instead start at {}".format(`
	`1049`	`+ raise DataError(`
	`1050`	`+ "fetch_results failed due to inconsistency in the state between the client and the server. Expected results to start from {} but they instead start at {}, some result batch must have been skipped".format(`
`1051`	`1051`	`expected_row_start_offset, resp.results.startRowOffset`
`1052`	`1052`	`)`
`1053`	`1053`	`)`