diff --git a/weave/legacy/tests/test_arrow.py b/weave/legacy/tests/test_arrow.py index 64b93b11193..b9966221b94 100644 --- a/weave/legacy/tests/test_arrow.py +++ b/weave/legacy/tests/test_arrow.py @@ -37,8 +37,8 @@ from weave.legacy.weave.ops_primitives import list_, make_list from .util import list_arrow_test_helpers as lath -from weave.legacy.weave.tests.util import tag_test_util as ttu -from weave.legacy.weave.tests.util import weavejs_ops +from weave.legacy.tests.util import tag_test_util as ttu +from weave.legacy.tests.util import weavejs_ops from . import test_wb _loading_builtins_token = context_state.set_loading_built_ins() diff --git a/weave/legacy/tests/test_arrow_awl.py b/weave/legacy/tests/test_arrow_awl.py index cef19769530..03e857e288f 100644 --- a/weave/legacy/tests/test_arrow_awl.py +++ b/weave/legacy/tests/test_arrow_awl.py @@ -8,7 +8,7 @@ from weave.legacy.weave.arrow.convert import to_arrow from weave.legacy.weave.language_features.tagging import tag_store, tagged_value_type -from weave.legacy.weave.tests.util.concrete_tagged_value import ( +from weave.legacy.tests.util.concrete_tagged_value import ( TaggedValue, concrete_from_tagstore, concrete_to_tagstore, diff --git a/weave/legacy/tests/test_arrow_topy.py b/weave/legacy/tests/test_arrow_topy.py index df5b5050453..59317cc3ce8 100644 --- a/weave/legacy/tests/test_arrow_topy.py +++ b/weave/legacy/tests/test_arrow_topy.py @@ -6,7 +6,7 @@ import weave from weave.legacy.weave import ops_arrow -from weave.legacy.weave.tests.util.concrete_tagged_value import ( +from weave.legacy.tests.util.concrete_tagged_value import ( TaggedValue, concrete_from_tagstore, concrete_to_tagstore, diff --git a/weave/legacy/tests/test_custom_types.py b/weave/legacy/tests/test_custom_types.py index 0269704ebbe..75a1aaeefdc 100644 --- a/weave/legacy/tests/test_custom_types.py +++ b/weave/legacy/tests/test_custom_types.py @@ -6,7 +6,7 @@ from weave.legacy.weave import ops_arrow from ... import errors -from weave.legacy.weave.tests.util import geom +from weave.legacy.tests.util import geom def test_mapped_method_on_custom_type(): diff --git a/weave/legacy/tests/test_execute_fast.py b/weave/legacy/tests/test_execute_fast.py index 1bfae6356a3..01d26d43b18 100644 --- a/weave/legacy/tests/test_execute_fast.py +++ b/weave/legacy/tests/test_execute_fast.py @@ -2,7 +2,7 @@ from weave.legacy.weave import dispatch, weave_internal from weave.legacy.weave import weave_types as types -from weave.legacy.weave.tests.util import weavejs_ops +from weave.legacy.tests.util import weavejs_ops def test_nested_weavejs_call(): diff --git a/weave/legacy/tests/test_list_arrow_compat.py b/weave/legacy/tests/test_list_arrow_compat.py index 07246b25c62..401a1ae7a5c 100644 --- a/weave/legacy/tests/test_list_arrow_compat.py +++ b/weave/legacy/tests/test_list_arrow_compat.py @@ -14,8 +14,8 @@ ) from weave.legacy.weave.ops_primitives import dict_, list_ -from weave.legacy.weave.tests.util import tag_test_util as ttu -from weave.legacy.weave.tests.util import list_arrow_test_helpers as lath +from weave.legacy.tests.util import tag_test_util as ttu +from weave.legacy.tests.util import list_arrow_test_helpers as lath def filter_fn(row) -> bool: diff --git a/weave/legacy/tests/test_op_behaviors.py b/weave/legacy/tests/test_op_behaviors.py index a9c83f94575..13618b63153 100644 --- a/weave/legacy/tests/test_op_behaviors.py +++ b/weave/legacy/tests/test_op_behaviors.py @@ -28,12 +28,12 @@ ) from ...legacy import registry_mem -from weave.legacy.weave.tests.util.concrete_tagged_value import ( +from weave.legacy.tests.util.concrete_tagged_value import ( TaggedValue, concrete_from_tagstore, concrete_to_tagstore, ) -from weave.legacy.weave.tests.util.op_specs import OP_TEST_SPECS, OpSpec, OpSpecTestCase +from weave.legacy.tests.util.op_specs import OP_TEST_SPECS, OpSpec, OpSpecTestCase def assert_equal_with_tags(node: graph.Node, v: typing.Any, expected: typing.Any): diff --git a/weave/legacy/tests/test_table_ops.py b/weave/legacy/tests/test_table_ops.py index 2eae1cd03fb..2994235eb37 100644 --- a/weave/legacy/tests/test_table_ops.py +++ b/weave/legacy/tests/test_table_ops.py @@ -16,7 +16,7 @@ from weave.legacy.weave import weave_types as types from weave.legacy.weave.ops_domain import table as table_ops -from weave.legacy.weave.tests.util import weavejs_ops +from weave.legacy.tests.util import weavejs_ops TABLE_TYPES = ["list", "pandas", "sql"] diff --git a/weave/legacy/tests/test_wb.py b/weave/legacy/tests/test_wb.py index 1608e0e01c2..b6cc5a11d5d 100644 --- a/weave/legacy/tests/test_wb.py +++ b/weave/legacy/tests/test_wb.py @@ -20,7 +20,7 @@ from .test_wb_domain_ops import assert_gql_str_equal from ...tests import fixture_fakewandb as fwb -from weave.legacy.weave.tests.util import weavejs_ops +from weave.legacy.tests.util import weavejs_ops file_path_response = { "project_518fa79465d8ffaeb91015dce87e092f": { diff --git a/weave/legacy/tests/test_weavify.py b/weave/legacy/tests/test_weavify.py index fece416e416..f29d63ce652 100644 --- a/weave/legacy/tests/test_weavify.py +++ b/weave/legacy/tests/test_weavify.py @@ -4,7 +4,7 @@ from weave.legacy.weave import weave_types as types from ...legacy import weavify -from weave.legacy.weave.tests.util import geom +from weave.legacy.tests.util import geom @pytest.mark.parametrize( diff --git a/weave/legacy/weave/ops_primitives/test_list.py b/weave/legacy/weave/ops_primitives/test_list.py index 072c9c88ddd..f2bf31f7cd4 100644 --- a/weave/legacy/weave/ops_primitives/test_list.py +++ b/weave/legacy/weave/ops_primitives/test_list.py @@ -10,7 +10,7 @@ tagged_value_type, ) from weave.legacy.weave.ops_primitives import dict, errors, list_, number, runs -from weave.legacy.weave.tests.util import geom, weavejs_ops +from weave.legacy.tests.util import geom, weavejs_ops def test_unnest(): diff --git a/weave/legacy/weave/ops_primitives/test_typeddict.py b/weave/legacy/weave/ops_primitives/test_typeddict.py index e16b423268e..6f57ee0ef8a 100644 --- a/weave/legacy/weave/ops_primitives/test_typeddict.py +++ b/weave/legacy/weave/ops_primitives/test_typeddict.py @@ -5,7 +5,7 @@ from weave.legacy.weave import box from weave.legacy.weave.arrow.arrow import ArrowWeaveListType from weave.legacy.weave.language_features.tagging import tag_store, tagged_value_type -from weave.legacy.weave.tests.util.list_arrow_test_helpers import ArrowNode +from weave.legacy.tests.util.list_arrow_test_helpers import ArrowNode pick_options = [ # Basic Pick diff --git a/weave/tests/fixture_fakewandb.py b/weave/tests/fixture_fakewandb.py index aa7c485bd64..4406f36613a 100644 --- a/weave/tests/fixture_fakewandb.py +++ b/weave/tests/fixture_fakewandb.py @@ -11,6 +11,7 @@ import wandb import weave +from weave.legacy.tests.util.tag_test_util import op_add_tag # Note: We're mocking out the whole io_service right now. This is too # high level and doesn't test the actual io implementation. We should @@ -22,7 +23,6 @@ WeaveWBArtifactByIDURI, WeaveWBArtifactURI, ) -from weave.legacy.weave.tests.util.tag_test_util import op_add_tag TEST_TABLE_ARTIFACT_PATH = "testdata/wb_artifacits/test_res_1fwmcd3q:v0" ABS_TEST_TABLE_ARTIFACT_PATH = os.path.abspath(TEST_TABLE_ARTIFACT_PATH)