Lint

langchain-ai · Aug 20, 2024 · 62b3745 · 62b3745
1 parent eb7f578
commit 62b3745
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/langsmith/client.py b/python/langsmith/client.py
@@ -2997,11 +2997,15 @@ def clone_public_dataset(
         )
         ds = source_client.read_shared_dataset(token_uuid)
         dataset_name = dataset_name or ds.name
-        if self.has_dataset(dataset_name=dataset_name):
+        try:
+            ds = self.read_dataset(dataset_name=dataset_name)
             logger.info(
                 f"Dataset {dataset_name} already exists in your tenant. Skipping."
             )
-            return
+            return str(ds.id)
+        except ls_utils.LangSmithNotFoundError:
+            pass
+
         try:
             # Fetch examples first
             examples = list(source_client.list_shared_examples(token_uuid))