weavify

wandb · Aug 22, 2024 · 2cb5288 · 2cb5288
1 parent 48994a0
commit 2cb5288
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/weave/legacy/decorator_op.py b/weave/legacy/decorator_op.py
@@ -105,7 +105,7 @@ def wrap(f: Callable[P, R]) -> Callable[P, R]:
             mutation=mutation,
         )
         if weavify:
-            from weave.weavify import op_to_weave_fn
+            from weave.legacy.weavify import op_to_weave_fn
 
             op.weave_fn = op_to_weave_fn(op)
 

diff --git a/weave/legacy/ops_arrow/vectorize.py b/weave/legacy/ops_arrow/vectorize.py
@@ -10,8 +10,8 @@
     errors,
     registry_mem,
     weave_internal,
-    weavify,
 )
+from weave.legacy import weavify
 from weave import weave_types as types
 from weave.query_api import op, use
 from weave.legacy import dispatch, graph, graph_debug, op_args, op_def

diff --git a/weave/weavify.py → weave/legacy/weavify.py b/weave/weavify.py → weave/legacy/weavify.py
@@ -2,8 +2,8 @@
 
 from weave.legacy import graph, op_args, val_const
 
-from . import errors, weave_internal
-from . import weave_types as types
+from .. import errors, weave_internal
+from .. import weave_types as types
 
 if typing.TYPE_CHECKING:
     from weave.legacy import op_def

diff --git a/weave/tests/legacy/test_weavify.py b/weave/tests/legacy/test_weavify.py
@@ -2,8 +2,9 @@
 
 from weave.legacy import graph, ops
 
-from ... import weave_internal, weavify
+from ... import weave_internal
 from ... import weave_types as types
+from ...legacy import weavify
 from .. import geom