From 224e62290f7172f99a03fe5d33d4a1b431916439 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Sun, 8 Dec 2024 16:57:16 -0800 Subject: [PATCH] kill unnecessarily large imports from telemetry init --- .../providers/inline/telemetry/meta_reference/telemetry.py | 6 ++---- llama_stack/providers/utils/telemetry/__init__.py | 3 --- 2 files changed, 2 insertions(+), 7 deletions(-) diff --git a/llama_stack/providers/inline/telemetry/meta_reference/telemetry.py b/llama_stack/providers/inline/telemetry/meta_reference/telemetry.py index 095591f9a..2e4a778e4 100644 --- a/llama_stack/providers/inline/telemetry/meta_reference/telemetry.py +++ b/llama_stack/providers/inline/telemetry/meta_reference/telemetry.py @@ -24,10 +24,8 @@ from llama_stack.providers.inline.telemetry.meta_reference.console_span_processo from llama_stack.providers.inline.telemetry.meta_reference.sqlite_span_processor import ( SQLiteSpanProcessor, ) -from llama_stack.providers.utils.telemetry import ( - SQLiteTraceStore, - TelemetryDatasetMixin, -) +from llama_stack.providers.utils.telemetry.dataset_mixin import TelemetryDatasetMixin +from llama_stack.providers.utils.telemetry.sqlite_trace_store import SQLiteTraceStore from llama_stack.apis.telemetry import * # noqa: F403 diff --git a/llama_stack/providers/utils/telemetry/__init__.py b/llama_stack/providers/utils/telemetry/__init__.py index 2d95a5dc5..756f351d8 100644 --- a/llama_stack/providers/utils/telemetry/__init__.py +++ b/llama_stack/providers/utils/telemetry/__init__.py @@ -3,6 +3,3 @@ # # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. - -from .dataset_mixin import TelemetryDatasetMixin # noqa: F401 -from .sqlite_trace_store import SQLiteTraceStore, TraceStore # noqa: F401