move to legacy: engine_trace

wandb · Aug 27, 2024 · a9bee1f · a9bee1f
1 parent 328d7cd
commit a9bee1f
Show file tree

Hide file tree

Showing 29 changed files with 96 additions and 92 deletions.
diff --git a/weave/filesystem.py b/weave/filesystem.py
@@ -20,9 +20,9 @@
 from aiofiles.threadpool import binary as aiofiles_binary
 from aiofiles.threadpool import text as aiofiles_text
 
-from weave.legacy import cache
+from weave.legacy import cache, engine_trace
 
-from . import engine_trace, environment, errors, util
+from . import environment, errors, util
 
 tracer = engine_trace.tracer()  # type: ignore
 async_utime = aiofiles_os.wrap(os.utime)  # type: ignore

diff --git a/weave/legacy/arrow/concat.py b/weave/legacy/arrow/concat.py
@@ -30,7 +30,7 @@
 
 DEBUG = False
 
-from weave import engine_trace
+from weave.legacy import engine_trace
 
 tracer = engine_trace.tracer()
 

diff --git a/weave/legacy/artifact_wandb.py b/weave/legacy/artifact_wandb.py
@@ -14,14 +14,14 @@
 from wandb.sdk.lib.hashutil import b64_to_hex_id, hex_to_b64_id
 
 from weave import (
-    engine_trace,
     errors,
     filesystem,
     urls,
     util,
 )
 from weave import weave_types as types
 from weave.legacy import (
+    engine_trace,
     artifact_fs,
     eager,
     file_base,

diff --git a/weave/legacy/cache.py b/weave/legacy/cache.py
@@ -6,8 +6,8 @@
 import time
 import typing
 
-from weave import engine_trace, environment, errors
-from weave.legacy import context_state, wandb_api
+from weave import environment, errors
+from weave.legacy import context_state, wandb_api, engine_trace
 
 statsd = engine_trace.statsd()  # type: ignore
 logger = logging.getLogger("root")

diff --git a/weave/legacy/compile.py b/weave/legacy/compile.py
@@ -6,12 +6,12 @@
 import typing
 
 from weave import (
-    engine_trace,
     errors,
     weave_internal,
 )
 from weave import weave_types as types
 from weave.legacy import (
+    engine_trace,
     box,
     compile_domain,
     compile_table,