|
1 | 1 | import time
|
2 | 2 | from typing import Dict, Tuple, List, Optional, Any, Union, Sequence
|
3 |
| -import uuid |
4 |
| -from databricks.sql.telemetry.TelemetryEvent import TelemetryEvent |
5 |
| -from databricks.sql.telemetry.DriverSystemConfiguration import DriverSystemConfiguration |
6 |
| -from databricks.sql.telemetry.TelemetryClient import TelemetryClient |
7 |
| -from databricks.sql.telemetry.NoopTelemetryClient import NoopTelemetryClient |
8 |
| -from databricks.sql.telemetry.TelemetryFrontendLog import TelemetryFrontendLog |
9 |
| -from databricks.sql.telemetry.FrontendLogContext import FrontendLogContext |
10 |
| -from databricks.sql.telemetry.TelemetryClientContext import TelemetryClientContext |
11 |
| -from databricks.sql.telemetry.FrontendLogEntry import FrontendLogEntry |
12 |
| -from databricks.sql.auth.auth import AuthType |
13 |
| -from databricks.sql.auth.authenticators import ( |
14 |
| - DatabricksOAuthProvider, |
15 |
| - ExternalAuthProvider, |
16 |
| - AuthProvider, |
17 |
| - AccessTokenAuthProvider, |
18 |
| -) |
19 |
| - |
20 | 3 | import pandas
|
21 | 4 |
|
22 | 5 | try:
|
@@ -257,25 +240,6 @@ def read(self) -> Optional[OAuthToken]:
|
257 | 240 | )
|
258 | 241 | telemetry_batch_size = kwargs.get("telemetry_batch_size", 200)
|
259 | 242 |
|
260 |
| - if self.telemetry_enabled: |
261 |
| - self.telemetry_client = TelemetryClient( |
262 |
| - host=self.host, |
263 |
| - connection_uuid="test-connection-uuid", |
264 |
| - auth_provider=auth_provider, |
265 |
| - is_authenticated=( |
266 |
| - isinstance(auth_provider, AccessTokenAuthProvider) |
267 |
| - or isinstance(auth_provider, DatabricksOAuthProvider) |
268 |
| - or isinstance(auth_provider, ExternalAuthProvider) |
269 |
| - or ( |
270 |
| - isinstance(auth_provider, AuthProvider) |
271 |
| - and hasattr(auth_provider, "_header_factory") |
272 |
| - ) |
273 |
| - ), |
274 |
| - batch_size=telemetry_batch_size, |
275 |
| - ) |
276 |
| - else: |
277 |
| - self.telemetry_client = NoopTelemetryClient() |
278 |
| - |
279 | 243 | user_agent_entry = kwargs.get("user_agent_entry")
|
280 | 244 | if user_agent_entry is None:
|
281 | 245 | user_agent_entry = kwargs.get("_user_agent_entry")
|
|
0 commit comments