From 2cb5288da2af5197021b1019649e2cc8379ae022 Mon Sep 17 00:00:00 2001
From: Andrew Truong <itsandrewtruong@gmail.com>
Date: Thu, 22 Aug 2024 00:28:52 -0400
Subject: [PATCH] weavify

---
 weave/legacy/decorator_op.py        | 2 +-
 weave/legacy/ops_arrow/vectorize.py | 2 +-
 weave/{ => legacy}/weavify.py       | 4 ++--
 weave/tests/legacy/test_weavify.py  | 3 ++-
 4 files changed, 6 insertions(+), 5 deletions(-)
 rename weave/{ => legacy}/weavify.py (98%)

diff --git a/weave/legacy/decorator_op.py b/weave/legacy/decorator_op.py
index d00e7f46ab4..6f27dcb142b 100644
--- 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
index 17bf1115b05..f4123b18a54 100644
--- 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 b/weave/legacy/weavify.py
similarity index 98%
rename from weave/weavify.py
rename to weave/legacy/weavify.py
index 271a9672694..e739a308e92 100644
--- a/weave/weavify.py
+++ b/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
index 5a2d0e5cda7..d869848d98d 100644
--- 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