cleanup: missing imports

wandb · Aug 27, 2024 · 420a5b9 · 420a5b9
1 parent bf01f16
commit 420a5b9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/weave/legacy/weave/ref_base.py b/weave/legacy/weave/ref_base.py
@@ -17,7 +17,7 @@
 
 if typing.TYPE_CHECKING:
     from weave.legacy.weave import weave_types as types
-    from ..trace import weave_client
+    from ...trace import weave_client
 
 
 def _map_to_ref_strs(obj: typing.Any) -> typing.Any:

diff --git a/weave/trace/box.py b/weave/trace/box.py
@@ -10,7 +10,7 @@
 
 import numpy as np
 
-from weave.legacy.weave.ref_base import Ref
+from weave.trace.refs import Ref
 
 T = TypeVar("T")
 

diff --git a/weave/trace/op_type.py b/weave/trace/op_type.py
@@ -19,11 +19,11 @@
     get_class_source,
     is_running_interactively,
 )
+from weave.trace.op import Op
 from weave.trace.refs import ObjectRef
 
 from ..legacy.weave import environment
 from . import serializer
-from .op import Op
 
 WEAVE_OP_PATTERN = re.compile(r"@weave\.op(\(\))?")
 WEAVE_OP_NO_PAREN_PATTERN = re.compile(r"@weave\.op(?!\()")