Skip to content

Commit

Permalink
delete: extraneous import
Browse files Browse the repository at this point in the history
  • Loading branch information
andrewtruong committed Aug 28, 2024
1 parent d7a84b1 commit f8f85ed
Show file tree
Hide file tree
Showing 5 changed files with 10 additions and 7 deletions.
1 change: 0 additions & 1 deletion weave/tests/trace/op_versioning_solo.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
import numpy as np

import weave
from weave.legacy.weave import artifact_fs


@weave.op()
Expand Down
2 changes: 1 addition & 1 deletion weave/tests/trace/test_op_versioning.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
import pytest

import weave
from weave.legacy.weave import artifact_fs, derive_op, op_def
from weave.legacy.weave import artifact_fs, derive_op
from weave.trace_server.trace_server_interface import FileContentReadReq, ObjReadReq


Expand Down
6 changes: 6 additions & 0 deletions weave/trace/errors.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,3 +17,9 @@ class WeaveOpSerializeError(WeaveSerializeError): ...


class WeaveInitError(Error): ...


class WeaveDefinitionError(Error): ...


class WeaveWandbAuthenticationException(Error): ...
5 changes: 2 additions & 3 deletions weave/trace/serve_fastapi.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,13 +12,12 @@
except ImportError:
from typing_extensions import Annotated # type: ignore

from weave.legacy.weave import cache, op_args, pyfunc_type_util
from weave.legacy.weave import cache, op_args, pyfunc_type_util, weave_pydantic
from weave.legacy.weave.wandb_api import WandbApiAsync
from weave.trace import errors
from weave.trace.op import Op
from weave.trace.refs import ObjectRef

from ..legacy.weave import errors, weave_pydantic

key_cache: cache.LruTimeWindowCache[str, typing.Optional[bool]] = (
cache.LruTimeWindowCache(datetime.timedelta(minutes=5))
)
Expand Down
3 changes: 1 addition & 2 deletions weave/trace/weave_init.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
import typing

from weave.legacy.weave import errors
from weave.trace import autopatch, init_message, trace_sentry, weave_client
from weave.trace import autopatch, errors, init_message, trace_sentry, weave_client
from weave.trace.client_context import weave_client as weave_client_context
from weave.trace_server import remote_http_trace_server, sqlite_trace_server

Expand Down

0 comments on commit f8f85ed

Please sign in to comment.