From 5f960af813a444af9836e844ba98b86491f345b5 Mon Sep 17 00:00:00 2001
From: Andrew Truong <itsandrewtruong@gmail.com>
Date: Thu, 22 Aug 2024 21:52:48 -0400
Subject: [PATCH] move to legacy: eager

---
 weave/legacy/artifact_wandb.py       | 2 +-
 weave/{ => legacy}/eager.py          | 4 ++--
 weave/legacy/execute.py              | 2 +-
 weave/legacy/run_streamtable_span.py | 2 +-
 4 files changed, 5 insertions(+), 5 deletions(-)
 rename weave/{ => legacy}/eager.py (98%)

diff --git a/weave/legacy/artifact_wandb.py b/weave/legacy/artifact_wandb.py
index 0d22e8b645a1..527d3c3c3aba 100644
--- 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 b/weave/legacy/eager.py
similarity index 98%
rename from weave/eager.py
rename to weave/legacy/eager.py
index 13c5ac76e85a..5136297e7beb 100644
--- a/weave/eager.py
+++ b/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
index 909a052edaa8..e05958a2f9c9 100644
--- 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
index 269187d8b0b1..ae5f9225c226 100644
--- 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