From 07806f85c7a51708e7e629a6953606f9e2ccbee1 Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Mon, 30 Dec 2024 14:16:28 -0800
Subject: [PATCH] clean up prints

---
 llama_stack/distribution/stack.py              | 2 --
 llama_stack/providers/tests/agents/fixtures.py | 2 --
 llama_stack/providers/tests/resolver.py        | 3 ---
 3 files changed, 7 deletions(-)

diff --git a/llama_stack/distribution/stack.py b/llama_stack/distribution/stack.py
index b5b87e579..965df5f03 100644
--- a/llama_stack/distribution/stack.py
+++ b/llama_stack/distribution/stack.py
@@ -185,8 +185,6 @@ async def construct_stack(
     impls = await resolve_impls(
         run_config, provider_registry or get_provider_registry(), dist_registry
     )
-    print("impls", impls)
-    print("run_config", run_config)
     await register_resources(run_config, impls)
     return impls
 
diff --git a/llama_stack/providers/tests/agents/fixtures.py b/llama_stack/providers/tests/agents/fixtures.py
index a2b927688..4159f18f4 100644
--- a/llama_stack/providers/tests/agents/fixtures.py
+++ b/llama_stack/providers/tests/agents/fixtures.py
@@ -8,7 +8,6 @@ import tempfile
 
 import pytest
 import pytest_asyncio
-from rich.pretty import pprint
 
 from llama_stack.apis.models import ModelInput, ModelType
 from llama_stack.distribution.datatypes import Api, Provider
@@ -99,7 +98,6 @@ async def agents_stack(request, inference_model, safety_shield):
         )
     )
 
-    pprint(f"fixture providers: {providers}")
     test_stack = await construct_stack_for_test(
         [Api.agents, Api.inference, Api.safety, Api.memory],
         providers,
diff --git a/llama_stack/providers/tests/resolver.py b/llama_stack/providers/tests/resolver.py
index 6b66060e1..5a38aaecc 100644
--- a/llama_stack/providers/tests/resolver.py
+++ b/llama_stack/providers/tests/resolver.py
@@ -9,7 +9,6 @@ import tempfile
 from typing import Any, Dict, List, Optional
 
 from pydantic import BaseModel
-from termcolor import cprint
 
 from llama_stack.apis.datasets import DatasetInput
 from llama_stack.apis.eval_tasks import EvalTaskInput
@@ -58,9 +57,7 @@ async def construct_stack_for_test(
         scoring_fns=scoring_fns or [],
         eval_tasks=eval_tasks or [],
     )
-    cprint(f"run_config BEFORE: {run_config}", "yellow")
     run_config = parse_and_maybe_upgrade_config(run_config)
-    cprint(f"run_config AFTER: {run_config}", "red")
     try:
         remote_config = remote_provider_config(run_config)
         if not remote_config: