arize 8.0.0a7__tar.gz → 8.0.0a9__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arize
-Version: 8.0.0a7
+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.0a7 → 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.0a7 → arize-8.0.0a9}/src/arize/_exporter/parsers/tracing_data_parser.py

@@ -1,8 +1,9 @@
-from __future__ import annotations
-
 import json
 import logging
-from typing import TYPE_CHECKING, List
+from typing import List
+
+import numpy as np
+import pandas as pd
 
 from arize.spans.columns import (
     SPAN_ATTRIBUTES_EMBEDDING_EMBEDDINGS_COL,
@@ -18,10 +19,6 @@ from arize.spans.columns import (
     SPAN_START_TIME_COL,
 )
 
-if TYPE_CHECKING:
-    import numpy as np
-    import pandas as pd
-
 logger = logging.getLogger(__name__)
 
 

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

@@ -9,6 +9,7 @@ from typing import TYPE_CHECKING, Any, Dict, List, Tuple
 from google.protobuf import json_format
 from pyarrow import flight
 
+from arize._flight.types import FlightRequestType
 from arize._generated.protocol.flight.ingest_pb2 import (
     WriteSpanAnnotationResponse,
     WriteSpanAttributesMetadataResponse,
@@ -22,7 +23,6 @@ from arize.version import __version__
 if TYPE_CHECKING:
     import pyarrow as pa
 
-    from arize._flight.client import FlightRequestType
 
 BytesPair = Tuple[bytes, bytes]
 Headers = List[BytesPair]

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

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

{arize-8.0.0a7 → 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.0a7 → arize-8.0.0a9}/src/arize/spans/validation/common/argument_validation.py

@@ -1,13 +1,10 @@
-from __future__ import annotations
+from typing import Any, List
 
-from typing import TYPE_CHECKING, Any, List
+import pandas as pd
 
 from arize.exceptions.base import InvalidFieldTypeConversion
 from arize.spans.validation.common.errors import InvalidTypeArgument
 
-if TYPE_CHECKING:
-    import pandas as pd
-
 
 def check_field_convertible_to_str(
     project_name: Any,

{arize-8.0.0a7 → arize-8.0.0a9}/src/arize/spans/validation/metadata/argument_validation.py

@@ -1,12 +1,9 @@
-from __future__ import annotations
+from typing import List
 
-from typing import TYPE_CHECKING, List
+import pandas as pd
 
 from arize.exceptions.base import ValidationError
 
-if TYPE_CHECKING:
-    import pandas as pd
-
 
 class MetadataArgumentError(ValidationError):
     def __init__(self, message: str, resolution: str) -> None:

{arize-8.0.0a7 → 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.0a7/src/arize/version.py

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