move to trace: rich_*

wandb · Aug 27, 2024 · f159a6d · f159a6d
1 parent 906ae55
commit f159a6d
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/weave/trace/feedback.py b/weave/trace/feedback.py
@@ -5,11 +5,11 @@
 
 from rich.table import Table
 
-from weave import rich_pydantic_util
 from weave.client_context import weave_client as weave_client_context
-from weave.refs import Refs
-from weave.rich_container import AbstractRichContainer
+from weave.trace import rich_pydantic_util
 from weave.trace.refs import parse_uri
+from weave.trace.rich_container import AbstractRichContainer
+from weave.trace.rich_refs import Refs
 from weave.trace_server import trace_server_interface as tsi
 from weave.trace_server.interface.query import Query
 

diff --git a/weave/rich_container.py → weave/trace/rich_container.py b/weave/rich_container.py → weave/trace/rich_container.py
@@ -3,7 +3,7 @@
 
 from rich.table import Table
 
-from weave import rich_pydantic_util
+from weave.trace import rich_pydantic_util
 
 T = TypeVar("T")
 

diff --git a/weave/rich_pydantic_util.py → weave/trace/rich_pydantic_util.py b/weave/rich_pydantic_util.py → weave/trace/rich_pydantic_util.py
@@ -6,7 +6,7 @@
 from rich.console import Console
 from rich.table import Table
 
-from . import util
+from .. import util
 
 
 def dict_to_table(d: dict[str, Any]) -> Table:

diff --git a/weave/refs.py → weave/trace/rich_refs.py b/weave/refs.py → weave/trace/rich_refs.py
@@ -5,8 +5,8 @@
 from rich.table import Table
 
 from weave.client_context import weave_client as weave_client_context
-from weave.rich_container import AbstractRichContainer
 from weave.trace.refs import AnyRef, CallRef, parse_uri
+from weave.trace.rich_container import AbstractRichContainer
 from weave.trace.vals import WeaveObject