arize 8.0.0a8__tar.gz → 8.0.0a9__tar.gz

{arize-8.0.0a8 → arize-8.0.0a9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize
-Version: 8.0.0a8
+Version: 8.0.0a9
 Summary: A helper library to interact with Arize AI APIs
 Project-URL: Homepage, https://arize.com
 Project-URL: Documentation, https://docs.arize.com/arize
@@ -34,6 +34,7 @@ Provides-Extra: ml-batch
 Requires-Dist: pandas<3,>=1.0.0; extra == 'ml-batch'
 Requires-Dist: protobuf<6,>=4.21.0; extra == 'ml-batch'
 Requires-Dist: pyarrow>=0.15.0; extra == 'ml-batch'
+Requires-Dist: requests<3,>=2.0.0; extra == 'ml-batch'
 Requires-Dist: tqdm; extra == 'ml-batch'
 Provides-Extra: ml-stream
 Requires-Dist: protobuf<6,>=4.21.0; extra == 'ml-stream'
@@ -44,6 +45,7 @@ Requires-Dist: opentelemetry-semantic-conventions<1,>=0.43b0; extra == 'spans'
 Requires-Dist: pandas<3,>=1.0.0; extra == 'spans'
 Requires-Dist: protobuf<6,>=4.21.0; extra == 'spans'
 Requires-Dist: pyarrow>=0.15.0; extra == 'spans'
+Requires-Dist: requests<3,>=2.0.0; extra == 'spans'
 Requires-Dist: tqdm; extra == 'spans'
 Description-Content-Type: text/markdown
 

{arize-8.0.0a8 → arize-8.0.0a9}/pyproject.toml

@@ -34,8 +34,8 @@ classifiers = [
     "Topic :: System :: Monitoring",
 ]
 dependencies = [
+    "numpy>=2.0.0", # For vector embeddings
     "lazy-imports",
-    "numpy>=2.0.0",
     # "requests_futures==1.0.0",
     # "googleapis_common_protos>=1.51.0,<2",
     # "protobuf>=4.21.0,<6",
@@ -57,7 +57,8 @@ spans = [
     "pandas>=1.0.0,<3",
     "protobuf>=4.21.0,<6",
     "pyarrow>=0.15.0",
-    "tqdm",
+    "requests>=2.0.0, <3", # For posting pyarrow files
+    "tqdm", # For export progress bars
 ]
 ml-stream = [
     "requests_futures>=1.0.0, <2",
@@ -65,9 +66,10 @@ ml-stream = [
 ]
 ml-batch = [
     "pandas>=1.0.0,<3",
-    "pyarrow>=0.15.0",
     "protobuf>=4.21.0,<6",
-    "tqdm",
+    "pyarrow>=0.15.0",
+    "requests>=2.0.0, <3", # For posting pyarrow files
+    "tqdm", # For export progress bars
 ]
 # datasets-experiments = [
 #     "pydantic",

{arize-8.0.0a8 → arize-8.0.0a9}/src/arize/client.py

@@ -65,6 +65,7 @@ class ArizeClient(LazySubclientsMixin):
                 "opentelemetry",
                 "pandas",
                 "pyarrow",
+                "requests",
                 "tqdm",
             ),
         ),

{arize-8.0.0a8 → arize-8.0.0a9}/src/arize/models/client.py

@@ -52,7 +52,6 @@ from arize.types import (
     is_list_of,
 )
 from arize.utils.casting import cast_dictionary, cast_typed_columns
-from arize.utils.dataframe import remove_extraneous_columns
 
 if TYPE_CHECKING:
     import concurrent.futures as cf
@@ -75,8 +74,9 @@ _STREAM_EXTRA = "ml-stream"
 
 _BATCH_DEPS = (
     "pandas",
-    "pyarrow",
     "google.protobuf",
+    "pyarrow",
+    "requests",
     "tqdm",
 )
 _BATCH_EXTRA = "ml-batch"
@@ -464,6 +464,7 @@ class MLModelsClient:
 
         from arize.models.batch_validation.validator import Validator
         from arize.utils.arrow import post_arrow_table
+        from arize.utils.dataframe import remove_extraneous_columns
         from arize.utils.proto import get_pb_schema, get_pb_schema_corpus
 
         # This method requires a space_id and project_name

{arize-8.0.0a8 → arize-8.0.0a9}/src/arize/utils/arrow.py

@@ -8,11 +8,12 @@ import tempfile
 from typing import TYPE_CHECKING, Any, Dict
 
 import pyarrow as pa
-import requests
 
 from arize.logging import get_arize_project_url, log_a_list
 
 if TYPE_CHECKING:
+    import requests
+
     from arize._generated.protocol.rec import public_pb2 as pb2
 
 logger = logging.getLogger(__name__)
@@ -27,6 +28,9 @@ def post_arrow_table(
     verify: bool,
     tmp_dir: str = "",
 ) -> requests.Response:
+    # We import here to avoid depending onn requests for all arrow utils
+    import requests
+
     logger.debug("Preparing to log Arrow table via file upload")
     logger.debug(
         "Preparing to log Arrow table via file upload",
@@ -71,15 +75,17 @@ def post_arrow_table(
             "Uploading file to Arize",
             extra={"path": outfile, "size_bytes": _filesize(outfile)},
         )
-        resp = _post_file(
-            files_url=files_url,
-            path=outfile,
-            headers=headers,
-            timeout=timeout,
-            verify=verify,
-        )
-        _maybe_log_project_url(resp)
-        return resp
+        # Post file
+        with open(outfile, "rb") as f:
+            resp = requests.post(
+                files_url,
+                timeout=timeout,
+                data=f,
+                headers=headers,
+                verify=verify,
+            )
+            _maybe_log_project_url(resp)
+            return resp
     finally:
         if tdir is not None:
             try:
@@ -98,23 +104,6 @@ def post_arrow_table(
                 )
 
 
-def _post_file(
-    files_url: str,
-    path: str,
-    headers: Dict[str, str],
-    timeout: float | None,
-    verify: bool,
-) -> requests.Response:
-    with open(path, "rb") as f:
-        return requests.post(
-            files_url,
-            timeout=timeout,
-            data=f,
-            headers=headers,
-            verify=verify,
-        )
-
-
 def append_to_pyarrow_metadata(
     pa_schema: pa.Schema, new_metadata: Dict[str, Any]
 ):

arize-8.0.0a9/src/arize/version.py

@@ -0,0 +1 @@
+__version__ = "8.0.0a9"

arize-8.0.0a8/src/arize/version.py

@@ -1 +0,0 @@
-__version__ = "8.0.0a8"