weavejs_fixes

wandb · Aug 22, 2024 · f9b17cf · f9b17cf
1 parent 2cb5288
commit f9b17cf
Show file tree

Hide file tree

Showing 7 changed files with 8 additions and 10 deletions.
diff --git a/weave/legacy/op_def.py b/weave/legacy/op_def.py
@@ -31,7 +31,7 @@
     tagged_value_type,
 )
 from weave.legacy.run import Run
-from weave.weavejs_fixes import fixup_node
+from weave.legacy.weavejs_fixes import fixup_node
 
 if typing.TYPE_CHECKING:
     from weave.trace import weave_client

diff --git a/weave/weavejs_fixes.py → weave/legacy/weavejs_fixes.py b/weave/weavejs_fixes.py → weave/legacy/weavejs_fixes.py
@@ -10,7 +10,7 @@
 
 from weave.legacy import graph
 
-from . import weave_types
+from .. import weave_types
 
 
 def _convert_specific_opname_to_generic_opname(

diff --git a/weave/show.py b/weave/show.py
@@ -14,9 +14,9 @@
     storage,
     usage_analytics,
     util,
-    weavejs_fixes,
 )
 from . import weave_types as types
+from .legacy import weavejs_fixes
 
 
 def make_varname_for_type(t: types.Type):

diff --git a/weave/tests/legacy/test_js_compat.py b/weave/tests/legacy/test_js_compat.py
@@ -8,8 +8,7 @@
 #     weave Python code that I haven't documented here.
 
 from weave import weave_types as types
-from weave import weavejs_fixes
-from weave.legacy import partial_object
+from weave.legacy import partial_object, weavejs_fixes
 from weave.legacy.ops_domain import wb_domain_types
 
 

diff --git a/weave/tests/legacy/test_storage.py b/weave/tests/legacy/test_storage.py
@@ -13,8 +13,8 @@
 from ... import api as weave
 from ... import storage
 from ... import weave_types as types
+from ...legacy.weavejs_fixes import recursively_unwrap_unions
 from ...weave_internal import make_const_node
-from ...weavejs_fixes import recursively_unwrap_unions
 from . import test_helpers
 
 

diff --git a/weave/tests/legacy/test_weavejs_fixes.py b/weave/tests/legacy/test_weavejs_fixes.py
@@ -2,9 +2,9 @@
 
 import pytest
 
-from weave.legacy import context_state, mappers_python, ops
+from weave.legacy import context_state, mappers_python, ops, weavejs_fixes
 
-from ... import api, weave_internal, weavejs_fixes
+from ... import api, weave_internal
 from ... import weave_types as types
 
 

diff --git a/weave/weave_server.py b/weave/weave_server.py
@@ -35,9 +35,8 @@
     server,
     storage,
     util,
-    weavejs_fixes,
 )
-from weave.legacy import context_state, graph, value_or_error, wandb_api
+from weave.legacy import context_state, graph, value_or_error, wandb_api, weavejs_fixes
 from weave.legacy.language_features.tagging import tag_store
 from weave.server_error_handling import client_safe_http_exceptions_as_werkzeug