move to legacy: weave_types

wandb · Aug 27, 2024 · 257446a · 257446a
1 parent fdffde0
commit 257446a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/weave/weave_pydantic.py → weave/legacy/weave_pydantic.py b/weave/weave_pydantic.py → weave/legacy/weave_pydantic.py
@@ -1,7 +1,7 @@
 from pydantic import BaseModel, create_model
 
-from . import weave_types as types
-from .legacy import infer_types
+from .. import weave_types as types
+from . import infer_types
 
 
 def weave_type_to_pydantic(

diff --git a/weave/serve_fastapi.py b/weave/serve_fastapi.py
@@ -17,7 +17,8 @@
 from weave.trace.op import Op
 from weave.trace.refs import ObjectRef
 
-from . import errors, weave_pydantic
+from . import errors
+from .legacy import weave_pydantic
 
 key_cache: cache.LruTimeWindowCache[str, typing.Optional[bool]] = (
     cache.LruTimeWindowCache(datetime.timedelta(minutes=5))

diff --git a/weave/weave_types.py b/weave/weave_types.py
@@ -1057,7 +1057,7 @@ def property_types(self) -> dict[str, Type]:
 
     @classmethod
     def typeclass_of_class(cls, check_class):
-        from . import weave_pydantic
+        from .legacy import weave_pydantic
 
         if not issubclass(check_class, pydantic.BaseModel):
             return cls