nomic 3.5.1__tar.gz → 3.5.2__tar.gz

{nomic-3.5.1 → nomic-3.5.2}/PKG-INFO

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

{nomic-3.5.1 → nomic-3.5.2}/nomic/dataset.py

@@ -700,32 +700,6 @@ class AtlasProjection:
     def tile_destination(self):
         return Path("~/.nomic/cache", self.id).expanduser()
 
-    def _get_atoms(self, ids: List[str]) -> List[Dict]:
-        """
-        Retrieves atoms by id
-
-        Args:
-            ids: list of atom ids
-
-        Returns:
-            A dictionary containing the resulting atoms, keyed by atom id.
-
-        """
-
-        if not isinstance(ids, list):
-            raise ValueError("You must specify a list of ids when getting data.")
-
-        response = requests.post(
-            self.dataset.atlas_api_path + "/v1/project/atoms/get",
-            headers=self.dataset.header,
-            json={"project_id": self.dataset.id, "index_id": self.atlas_index_id, "atom_ids": ids},
-        )
-
-        if response.status_code == 200:
-            return response.json()["atoms"]
-        else:
-            raise Exception(response.text)
-
 
 class AtlasDataStream(AtlasClass):
     def __init__(self, name: Optional[str] = "contrastors"):
@@ -1445,7 +1419,7 @@ class AtlasDataset(AtlasClass):
                 images.append((current_temp_id, processed_blob_value))
 
         batch_size = 40
-        num_workers = 10
+        num_workers = 2
 
         def send_request(batch_start_index):
             image_batch = images[batch_start_index : batch_start_index + batch_size]
@@ -1592,12 +1566,8 @@ class AtlasDataset(AtlasClass):
             None
         """
 
-        # Exactly 10 upload workers at a time.
-
-        num_workers = 10
-
+        num_workers = 2
         # Each worker currently is too slow beyond a shard_size of 10000
-
         # The heuristic here is: Never let shards be more than 10,000 items,
         # OR more than 16MB uncompressed. Whichever is smaller.
 
@@ -1713,7 +1683,7 @@ class AtlasDataset(AtlasClass):
             else:
                 logger.info("Upload succeeded.")
 
-    def update_maps(self, data: List[Dict], embeddings: Optional[np.ndarray] = None, num_workers: int = 10):
+    def update_maps(self, data: List[Dict], embeddings: Optional[np.ndarray] = None, num_workers: int = 2):
         """
         Utility method to update a project's maps by adding the given data.
 

{nomic-3.5.1 → nomic-3.5.2}/nomic.egg-info/PKG-INFO

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

{nomic-3.5.1 → nomic-3.5.2}/setup.py

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