move to legacy: eager

wandb · Aug 27, 2024 · 5f960af · 5f960af
1 parent 83632e7
commit 5f960af
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/weave/legacy/artifact_wandb.py b/weave/legacy/artifact_wandb.py
@@ -14,7 +14,6 @@
 from wandb.sdk.lib.hashutil import b64_to_hex_id, hex_to_b64_id
 
 from weave import (
-    eager,
     engine_trace,
     errors,
     filesystem,
@@ -24,6 +23,7 @@
 from weave import weave_types as types
 from weave.legacy import (
     artifact_fs,
+    eager,
     file_base,
     file_util,
     memo,

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

diff --git a/weave/legacy/execute.py b/weave/legacy/execute.py
@@ -10,7 +10,6 @@
 from collections.abc import Mapping
 
 from weave import (
-    eager,
     engine_trace,
     environment,
     errors,
@@ -32,6 +31,7 @@
     compile,
     context,
     context_state,
+    eager,
     forward_graph,
     graph,
     graph_debug,

diff --git a/weave/legacy/run_streamtable_span.py b/weave/legacy/run_streamtable_span.py
@@ -2,7 +2,7 @@
 from typing import Iterable
 
 from weave.legacy import stream_data_interfaces
-from weave.eager import WeaveIter
+from weave.legacy.eager import WeaveIter
 from weave.legacy import artifact_wandb, uris
 from weave.legacy.run import Run