openprotein-python 0.8.12__tar.gz → 0.9.0__tar.gz

{openprotein_python-0.8.12 → openprotein_python-0.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openprotein-python
-Version: 0.8.12
+Version: 0.9.0
 Summary: OpenProtein Python interface.
 Author-email: Mark Gee <markgee@ne47.bio>, "Timothy Truong Jr." <ttruong@ne47.bio>, Tristan Bepler <tbepler@ne47.bio>
 License-Expression: MIT

{openprotein_python-0.8.12 → openprotein_python-0.9.0}/openprotein/embeddings/future.py

@@ -1,7 +1,7 @@
 """Future for embeddings-related jobs."""
 
 from collections import namedtuple
-from typing import Generator
+from typing import Any, Generator
 
 import numpy as np
 
@@ -45,10 +45,13 @@ class EmbeddingsResultFuture(MappedFuture, Future):
             else sequences
         )
 
-    def stream(self):
-        return api.request_get_embeddings_stream(session=self.session, job_id=self.id)
+    def stream_sync(self) -> Generator[tuple[str | bytes, np.ndarray], Any, None]:
+        for i, array in enumerate(
+            api.request_get_embeddings_stream(session=self.session, job_id=self.id)
+        ):
+            yield self.sequences[i], array
 
-    def get(self, verbose=False) -> list[np.ndarray]:
+    def get(self, verbose: bool = False) -> list[tuple[str | bytes, np.ndarray]]:
         return super().get(verbose=verbose)
 
     @property
@@ -74,7 +77,7 @@ class EmbeddingsResultFuture(MappedFuture, Future):
         """
         return self.sequences
 
-    def get_item(self, sequence: bytes) -> np.ndarray:
+    def get_item(self, sequence: str | bytes) -> np.ndarray:
         """
         Get embedding results for specified sequence.