nomic 3.3.2__tar.gz → 3.3.3__tar.gz

{nomic-3.3.2 → nomic-3.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nomic
-Version: 3.3.2
+Version: 3.3.3
 Summary: The official Nomic python client.
 Home-page: https://github.com/nomic-ai/nomic
 Author: nomic.ai

{nomic-3.3.2 → nomic-3.3.3}/nomic/dataset.py

@@ -12,6 +12,7 @@ from pathlib import Path
 from typing import Dict, List, Optional, Tuple, Union
 
 import numpy as np
+import pandas as pd
 import pyarrow as pa
 import requests
 from loguru import logger
@@ -1373,6 +1374,18 @@ class AtlasDataset(AtlasClass):
             blobs: A list of image paths, bytes, or PIL Images. Use if you want to create an AtlasDataset using image embeddings over your images. Note: Blobs are stored locally only.
             pbar: (Optional). A tqdm progress bar to update.
         """
+        if isinstance(data, DataFrame):
+            cols_before = set(data.columns)
+            for col in cols_before:
+                if col.startswith("_"):
+                    raise ValueError(
+                        f"You are attempting to upload a pandas dataframe with the column name {col}, but columns beginning with '_' are reserved for Atlas internal use. Please rename your column and try again."
+                    )
+            data = pa.Table.from_pandas(data)
+            for newcol in set(data.column_names).difference(cols_before):
+                logger.warning(f"Dropping column {newcol} added in pandas conversion to pyarrow")
+                data = data.drop([newcol])
+
         if embeddings is not None:
             self._add_embeddings(data=data, embeddings=embeddings, pbar=pbar)
         elif isinstance(data, pa.Table) and "_embeddings" in data.column_names:  # type: ignore

{nomic-3.3.2 → nomic-3.3.3}/nomic.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nomic
-Version: 3.3.2
+Version: 3.3.3
 Summary: The official Nomic python client.
 Home-page: https://github.com/nomic-ai/nomic
 Author: nomic.ai

{nomic-3.3.2 → nomic-3.3.3}/setup.py

@@ -23,7 +23,7 @@ with open("README.md") as f:
 
 setup(
     name="nomic",
-    version="3.3.2",
+    version="3.3.3",
     url="https://github.com/nomic-ai/nomic",
     description=description,
     long_description=long_description,