From 6065891682c77632c7af6c92d2e1c838ac3b99a4 Mon Sep 17 00:00:00 2001 From: Alexander Barannikov Date: Tue, 2 Jul 2024 17:12:01 +0100 Subject: [PATCH] [CM-10413] change import to avoid pyright linter error --- src/comet_llm/api_objects/api.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/comet_llm/api_objects/api.py b/src/comet_llm/api_objects/api.py index a383d1ff..f67f3adb 100644 --- a/src/comet_llm/api_objects/api.py +++ b/src/comet_llm/api_objects/api.py @@ -15,6 +15,7 @@ from typing import List, Optional import comet_ml +from comet_ml.query import QueryExpression from .. import experiment_info, logging_messages, query_dsl from . import llm_trace_api @@ -91,7 +92,7 @@ def get_llm_trace_by_name( return llm_trace_api.LLMTraceAPI.__api__from_api_experiment__(matching_trace[0]) def query( - self, workspace: str, project_name: str, query: comet_ml.QueryExpression + self, workspace: str, project_name: str, query: QueryExpression ) -> List[llm_trace_api.LLMTraceAPI]: """ Fetch LLM Trace based on a query. Currently, it is only possible to use