move to legacy: node_ref

wandb · Aug 27, 2024 · 865fe43 · 865fe43
1 parent ad8929d
commit 865fe43
Show file tree

Hide file tree

Showing 7 changed files with 10 additions and 9 deletions.
diff --git a/weave/legacy/arrow/list_.py b/weave/legacy/arrow/list_.py
@@ -12,7 +12,6 @@
 
 from weave import (
     errors,
-    node_ref,
     ref_base,
     ref_util,
     weave_internal,
@@ -27,6 +26,7 @@
     graph,
     op_def,
     op_def_type,
+    node_ref,
 )
 from weave.legacy.arrow.arrow import (
     ArrowWeaveListType,

diff --git a/weave/legacy/graph_mapper.py b/weave/legacy/graph_mapper.py
@@ -1,6 +1,6 @@
-from weave import node_ref, ref_base
+from weave import ref_base
 from weave import weave_types as types
-from weave.legacy import graph, mappers
+from weave.legacy import graph, mappers, node_ref
 from weave.legacy import mappers_python_def as mappers_python
 
 

diff --git a/weave/legacy/mappers_arrow.py b/weave/legacy/mappers_arrow.py
@@ -5,14 +5,15 @@
 
 import pyarrow as pa
 
-from weave import errors, node_ref, ref_base
+from weave import errors, ref_base
 from weave import weave_types as types
 from weave.legacy import (
     arrow_util,
     artifact_base,
     box,
     mappers,
     mappers_weave,
+    node_ref,
     partial_object,
 )
 from weave.legacy import mappers_python_def as mappers_python

diff --git a/weave/legacy/mappers_publisher.py b/weave/legacy/mappers_publisher.py
@@ -4,7 +4,7 @@
 
 from weave import errors, ref_base, storage, weave_internal
 from weave import weave_types as types
-from weave.node_ref import ref_to_node
+from weave.legacy.node_ref import ref_to_node
 
 # from weave.legacy.ops_primitives import weave_api
 from weave.legacy import (

diff --git a/weave/node_ref.py → weave/legacy/node_ref.py b/weave/node_ref.py → weave/legacy/node_ref.py
@@ -3,7 +3,7 @@
 
 from weave.legacy import graph
 
-from . import ref_base
+from .. import ref_base
 
 # Notes for the future:
 # - I added list.lookup to lookup rows in a list by ID. I think we probably should

diff --git a/weave/legacy/show.py b/weave/legacy/show.py
@@ -5,8 +5,8 @@
 
 from IPython.display import IFrame, display
 
-from weave.legacy import artifact_fs, context, graph, ops, panel
-from weave import errors, node_ref, ref_base, storage, util
+from weave.legacy import artifact_fs, context, graph, ops, node_ref, panel
+from weave import errors, ref_base, storage, util
 from . import usage_analytics
 from .. import weave_types as types
 from . import weavejs_fixes

diff --git a/weave/tests/legacy/test_node_ref.py b/weave/tests/legacy/test_node_ref.py
@@ -1,7 +1,7 @@
 from weave.legacy import graph
 
 from ... import api as weave
-from ... import node_ref
+from ...legacy import node_ref
 
 
 def test_node_to_ref():