symphony/client: added rich dependency

symphony/client/pyproject.toml

@@ -42,4 +42,5 @@ packages = [
 [tool.poetry.dependencies]
 aiohttp = "3.9.5"
 pydantic-settings = "2.6.1"
+rich = "13.9.4"
 simbricks-orchestration = "*"

symphony/client/requirements.txt

@@ -23,6 +23,7 @@
 # external dependencies
 aiohttp==3.9.5
 pydantic-settings==2.6.1
+rich==13.9.4
 
 # local dependencies
 -e ../orchestration
\ No newline at end of file

symphony/client/simbricks/client/client.py

@@ -25,13 +25,12 @@ import aiohttp
 import typing
 import contextlib
 import asyncio
-import rich
 import json
+from rich.console import Console
 from .auth import TokenProvider
 from .settings import client_settings
 from simbricks.orchestration import system
 from simbricks.orchestration import simulation
-from simbricks.orchestration import instantiation
 
 
 class BaseClient:
@@ -326,7 +325,7 @@ class SimBricksClient:
             return await resp.json()
 
     async def follow_run(self, run_id: int) -> None:
-        console = rich.console.Console()
+        console = Console()
         with console.status(f"[bold green]Waiting for run {run_id} to finish...") as status:
             last_run = None
             prev_len = 0