move to legacy: weave_internal

wandb · Aug 27, 2024 · 86b5b61 · 86b5b61
1 parent 8b7d953
commit 86b5b61
Show file tree

Hide file tree

Showing 97 changed files with 304 additions and 316 deletions.
diff --git a/examples/experimental/skip_test/Confusion Matrix.ipynb b/examples/experimental/skip_test/Confusion Matrix.ipynb
@@ -13,7 +13,7 @@
     "from PIL import Image\n",
     "import weave\n",
     "from weave.legacy import storage\n",
-    "from weave import weave_internal\n",
+    "from weave.legacy import weave_internal\n",
     "#weave.use_frontend_devmode()"
    ]
   },

diff --git a/examples/experimental/skip_test/Diffusion explore 2.ipynb b/examples/experimental/skip_test/Diffusion explore 2.ipynb
@@ -8,7 +8,7 @@
    "outputs": [],
    "source": [
     "import weave\n",
-    "from weave import weave_internal\n",
+    "from weave.legacy import weave_internal\n",
     "from weave.legacy.ecosystem.all import langchain\n",
     "weave.use_frontend_devmode()"
    ]

diff --git a/examples/experimental/skip_test/Diffusion explore.ipynb b/examples/experimental/skip_test/Diffusion explore.ipynb
@@ -25,7 +25,7 @@
    "outputs": [],
    "source": [
     "import weave\n",
-    "from weave import weave_internal\n",
+    "from weave.legacy import weave_internal\n",
     "from weave.legacy.ecosystem.all import langchain"
    ]
   },

diff --git a/examples/experimental/skip_test/Diffusion story.ipynb b/examples/experimental/skip_test/Diffusion story.ipynb
@@ -8,7 +8,7 @@
    "outputs": [],
    "source": [
     "import weave\n",
-    "from weave import weave_internal\n",
+    "from weave.legacy import weave_internal\n",
     "from weave.legacy.ecosystem.all import langchain\n",
     "weave.use_frontend_devmode()"
    ]

diff --git a/examples/experimental/skip_test/art_explore.ipynb b/examples/experimental/skip_test/art_explore.ipynb
@@ -32,7 +32,7 @@
    "outputs": [],
    "source": [
     "import weave\n",
-    "from weave import weave_internal\n",
+    "from weave.legacy import weave_internal\n",
     "from weave.legacy.ecosystem.all import langchain"
    ]
   },

diff --git a/examples/experimental/skip_test/weave_scifi.ipynb b/examples/experimental/skip_test/weave_scifi.ipynb
@@ -8,7 +8,7 @@
    "outputs": [],
    "source": [
     "import weave\n",
-    "from weave import weave_internal\n",
+    "from weave.legacy import weave_internal\n",
     "from weave.legacy.ecosystem.all import langchain\n",
     "weave.use_frontend_devmode()"
    ]

diff --git a/examples/experimental/tag_search.ipynb b/examples/experimental/tag_search.ipynb
@@ -30,7 +30,7 @@
    "outputs": [],
    "source": [
     "import weave\n",
-    "from weave import weave_internal"
+    "from weave.legacy import weave_internal"
    ]
   },
   {

diff --git a/examples/reference/panels/plot/synced_axes.ipynb b/examples/reference/panels/plot/synced_axes.ipynb
@@ -89,7 +89,7 @@
    "outputs": [],
    "source": [
     "from weave.legacy.panels.panel_plot import Signals, AxisSelections, Plot\n",
-    "from weave import weave_internal\n",
+    "from weave.legacy import weave_internal\n",
     "\n",
     "\n",
     "domain_node = weave_internal.const(None)\n",

diff --git a/examples/reference/vis/Distribution.ipynb b/examples/reference/vis/Distribution.ipynb
@@ -41,7 +41,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from weave import weave_internal\n",
+    "from weave.legacy import weave_internal\n",
     "# Numeric distribution\n",
     "dist = wandb.Distribution(\n",
     "    items,\n",

diff --git a/weave/legacy/api.py b/weave/legacy/api.py
@@ -12,7 +12,7 @@
 from . import ref_base as _ref_base
 from weave.legacy import wandb_api as _wandb_api
 
-from .. import weave_internal as _weave_internal
+from weave.legacy import weave_internal as _weave_internal
 
 from weave.legacy import util as _util
 

diff --git a/weave/legacy/arrow/convert.py b/weave/legacy/arrow/convert.py
@@ -4,7 +4,7 @@
 import pyarrow.compute as pc
 
 from weave.legacy import api as api
-from weave import weave_internal
+from weave.legacy import weave_internal
 from weave.legacy import weave_types as types
 from weave.legacy import errors, arrow_util, artifact_base, artifact_mem, box, mappers_arrow
 from weave.legacy.arrow.arrow import (

diff --git a/weave/legacy/arrow/list_.py b/weave/legacy/arrow/list_.py
@@ -10,11 +10,9 @@
 import pyarrow.compute as pc
 import typing_extensions
 
-from weave import (
-    weave_internal,
-)
 from weave.legacy import weave_types as types
 from weave.legacy import (
+    weave_internal,
     errors,
     _dict_utils,
     artifact_base,

diff --git a/weave/legacy/codify.py b/weave/legacy/codify.py
@@ -201,15 +201,15 @@ def _node_to_code(node: graph.Node, wrap_const_node: bool = True) -> str:
 
         if current_frame is not None and node.name in current_frame:
             return node.name
-        return f"weave.weave_internal.make_var_node({_type_to_code(node.type)}, '{node.name}')"
+        return f"weave.legacy.weave_internal.make_var_node({_type_to_code(node.type)}, '{node.name}')"
     elif isinstance(node, graph.ConstNode):
         if isinstance(node.type, weave_types.Function):
             vars = list(node.type.input_types.keys())
             return lambda_wrapped_object_to_code_no_format(node.val, vars)
         else:
             val_as_code = object_to_code_no_format(node.val)
             if wrap_const_node:
-                return f"weave.weave_internal.const({val_as_code})"
+                return f"weave.legacy.weave_internal.const({val_as_code})"
             else:
                 return val_as_code
     elif isinstance(node, graph.OutputNode):

diff --git a/weave/legacy/compile.py b/weave/legacy/compile.py
@@ -5,11 +5,9 @@
 import re
 import typing
 
-from weave import (
-    weave_internal,
-)
 from weave.legacy import weave_types as types
 from weave.legacy import (
+    weave_internal,
     errors,
     engine_trace,
     box,

diff --git a/weave/legacy/derive_op.py b/weave/legacy/derive_op.py
@@ -6,12 +6,10 @@
 import inspect
 import typing
 
-from weave import (
-    storage,
-    weave_internal,
-)
 from weave.legacy import weave_types as types
 from weave.legacy import (
+    storage,
+    weave_internal,
     errors,
     parallelism,
     registry_mem,

diff --git a/weave/legacy/eager.py b/weave/legacy/eager.py
@@ -2,7 +2,7 @@
 
 from weave.legacy import context_state, graph
 
-from .. import weave_internal
+from weave.legacy import weave_internal
 from weave.legacy import weave_types as types
 
 WeaveIterObjectType = typing.TypeVar("WeaveIterObjectType")

diff --git a/weave/legacy/ecosystem/wandb/panel_distribution.py b/weave/legacy/ecosystem/wandb/panel_distribution.py
@@ -2,7 +2,7 @@
 import typing
 
 import weave
-from weave import weave_internal
+from weave.legacy import weave_internal
 from weave.legacy import panel_util
 from weave.legacy.ecosystem.wandb import weave_plotly
 

diff --git a/weave/legacy/ecosystem/wandb/panel_geo.py b/weave/legacy/ecosystem/wandb/panel_geo.py
@@ -3,7 +3,7 @@
 import typing
 
 import weave
-from weave import weave_internal
+from weave.legacy import weave_internal
 from weave.legacy.ecosystem.wandb import weave_plotly
 
 

diff --git a/weave/legacy/ecosystem/wandb/panel_scatter.py b/weave/legacy/ecosystem/wandb/panel_scatter.py
@@ -3,7 +3,7 @@
 import typing
 
 import weave
-from weave import weave_internal
+from weave.legacy import weave_internal
 from weave.legacy.ecosystem.wandb import weave_plotly
 
 

diff --git a/weave/legacy/ecosystem/wandb/panel_time_series.py b/weave/legacy/ecosystem/wandb/panel_time_series.py
@@ -2,7 +2,7 @@
 import typing
 
 import weave
-from weave import weave_internal
+from weave.legacy import weave_internal
 from weave.legacy.ecosystem.wandb import weave_plotly
 from weave.legacy.language_features.tagging import tagged_value_type
 from weave.legacy.panel_util import make_node

diff --git a/weave/legacy/ecosystem/wandb/test_panel_distribution.py b/weave/legacy/ecosystem/wandb/test_panel_distribution.py
@@ -1,7 +1,7 @@
 import random
 
 import weave
-from weave import weave_internal
+from weave.legacy import weave_internal
 from weave.legacy.ecosystem.wandb import panel_distribution
 
 

diff --git a/weave/legacy/ecosystem/wandb/weave_plotly.py b/weave/legacy/ecosystem/wandb/weave_plotly.py
@@ -9,7 +9,7 @@
 from plotly import graph_objs as go
 
 import weave
-from weave import weave_internal
+from weave.legacy import weave_internal
 from weave.legacy import infer_types
 
 

diff --git a/weave/legacy/execute_fast.py b/weave/legacy/execute_fast.py
@@ -1,10 +1,8 @@
 import logging
 
-from weave import (
-    weave_internal,
-)
 from weave.legacy import weave_types as types
 from weave.legacy import (
+    weave_internal,
     errors,
     engine_trace,
     ref_base,

diff --git a/weave/legacy/make_type.py b/weave/legacy/make_type.py
@@ -1,4 +1,4 @@
-from .. import weave_internal
+from weave.legacy import weave_internal
 from weave.legacy import weave_types as types
 
 

diff --git a/weave/legacy/mappers_publisher.py b/weave/legacy/mappers_publisher.py
@@ -3,7 +3,7 @@
 import typing
 
 from weave.legacy import storage
-from weave import weave_internal
+from weave.legacy import weave_internal
 from weave.legacy import weave_types as types
 from weave.legacy.node_ref import ref_to_node