From 8c71e3b0199f04095d81b43c0bd77e768b7512cf Mon Sep 17 00:00:00 2001 From: mike0sv Date: Thu, 11 Jul 2024 18:14:06 +0400 Subject: [PATCH] fix tracing mypy --- src/evidently/tracing/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/evidently/tracing/__init__.py b/src/evidently/tracing/__init__.py index 2949cecf15..304eef49c3 100644 --- a/src/evidently/tracing/__init__.py +++ b/src/evidently/tracing/__init__.py @@ -13,6 +13,7 @@ from opentelemetry.sdk.trace import TracerProvider from opentelemetry.sdk.trace.export import BatchSpanProcessor from opentelemetry.sdk.trace.export import SpanExporter +from requests import Response from evidently.ui.workspace.cloud import ACCESS_TOKEN_COOKIE from evidently.ui.workspace.cloud import CloudMetadataStorage @@ -107,14 +108,15 @@ def _create_tracer_provider( ) cloud = CloudMetadataStorage(_address, _api_key, ACCESS_TOKEN_COOKIE.key) - datasets = cloud._request("/api/datasets", "GET").json()["datasets"] + datasets_response: Response = cloud._request("/api/datasets", "GET") + datasets = datasets_response.json()["datasets"] _export_id = None for dataset in datasets: if dataset["name"] == _export_name: _export_id = dataset["id"] break if _export_id is None: - resp = cloud._request( + resp: Response = cloud._request( "/api/datasets/tracing", "POST", query_params={"team_id": _team_id},