fix: fix import bug

weavel-ai · Sep 11, 2024 · 9d97189 · 9d97189
1 parent e82fc88
commit 9d97189
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
 
 setup(
     name="weavel",
-    version="1.9.0",
+    version="1.9.1",
     packages=find_namespace_packages(),
     entry_points={},
     description="Weavel, Prompt Optimization and Evaluation for LLM Applications",

diff --git a/weavel/client.py b/weavel/client.py
@@ -13,14 +13,13 @@
 from dotenv import load_dotenv
 from weavel._worker import Worker
 
-# from weavel.types.instances import Session, Span, Trace
 from weavel.object_clients import (
     GenerationClient,
     SessionClient,
     SpanClient,
     TraceClient,
 )
-from weavel.types.datasets import Dataset, DatasetItem, Prompt, PromptVersion
+from weavel.types import Dataset, DatasetItem, Prompt, PromptVersion
 
 load_dotenv()