e2e test fix

saishreeeee · saishreeeee · commit 12ff47d217b2 · 2025-07-19T10:42:59.000+05:30
Signed-off-by: Sai Shree Pradhan &lt;saishree.pradhan@databricks.com&gt;
diff --git a/tests/e2e/test_telemetry_retry.py b/tests/e2e/test_telemetry_retry.py
@@ -59,7 +59,7 @@ def get_client(self, session_id, num_retries=3):
             force_dangerous_codes=[],
             urllib3_kwargs={'total': num_retries}
         )
-        adapter = client._session.adapters.get("https://")
+        adapter = client._http_client.session.adapters.get("https://")
         adapter.max_retries = retry_policy
         return client, adapter
 

Original file line number	Diff line number	Diff line change
`@@ -59,7 +59,7 @@ def get_client(self, session_id, num_retries=3):`
`59`	`59`	`force_dangerous_codes=[],`
`60`	`60`	`urllib3_kwargs={'total': num_retries}`
`61`	`61`	`)`
`62`		`- adapter = client._session.adapters.get("https://")`
	`62`	`+ adapter = client._http_client.session.adapters.get("https://")`
`63`	`63`	`adapter.max_retries = retry_policy`
`64`	`64`	`return client, adapter`
`65`	`65`