|
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