arize 8.0.0a7__py3-none-any.whl → 8.0.0a9__py3-none-any.whl

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/_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/client.py

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

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/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/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/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/version.py

@@ -1 +1 @@
-__version__ = "8.0.0a7"
+__version__ = "8.0.0a9"

{arize-8.0.0a7.dist-info → arize-8.0.0a9.dist-info}/METADATA

@@ -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.dist-info → arize-8.0.0a9.dist-info}/RECORD

@@ -1,17 +1,17 @@
 arize/__init__.py,sha256=-4bbbZwcjGS9OfAunsB-lmKRCzccPdFvZmvJQJEky3E,534
 arize/_lazy.py,sha256=MVep6D93sJWvArg4pgm4CVNGc6tu-XRK_Z7EDMuc76I,2358
-arize/client.py,sha256=ntSYQSzM6MrvcYDxV-lFF-yoxoPjjCvBHKoN-2k1BGg,5688
+arize/client.py,sha256=0LtZU3WeEatGd1QgQsMrJOuI-tFmzM3y1AfO74BLJys,5716
 arize/config.py,sha256=iynVEZhrOPdTNJTQ_KQmwKOPiwL0LfEP8AUIDYW86Xw,5801
 arize/logging.py,sha256=2vwdta2-kR78GeBFGK2vpk51rQ2d06HoKzuARI9qFQk,7317
 arize/types.py,sha256=z1yg5-brmTD4kVHDmmTVkYke53JpusXXeOOpdQw7rYg,69508
-arize/version.py,sha256=QcLp83_rc-TQ4KtJaloeQDRJflEN04D3voc6Lf15giw,24
+arize/version.py,sha256=BDT_fj8aJ4pvQMjTUrdbEd6vY71GkAguJScCzty97gA,24
 arize/_exporter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arize/_exporter/client.py,sha256=eAxJX1sUfdpLrtaQ0ynMTd5jI37JOp9fbl3NWp4WFEA,15216
 arize/_exporter/validation.py,sha256=6ROu5p7uaolxQ93lO_Eiwv9NVw_uyi3E5T--C5Klo5Q,1021
 arize/_exporter/parsers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arize/_exporter/parsers/tracing_data_parser.py,sha256=1ZSYVlrpYYkewg-1QWxwR5aZMX8oglw1wq7kJlh0SkI,5472
+arize/_exporter/parsers/tracing_data_parser.py,sha256=zVS-w8t1HJkz-AIC_JCdjPJ7gJXgFpfELfqNM_vK42E,5395
 arize/_flight/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arize/_flight/client.py,sha256=aJzN584nlCPgPtGIh6BFYRogB6TvSlhRRmQcDdTehCY,6900
+arize/_flight/client.py,sha256=OCqMf0SEclc2TLzMqu3wgORuLHHGkZOODDZsUu8F8a0,6895
 arize/_flight/types.py,sha256=OuLupzkGYt7r0PEzsX4NmXV9uq3lD11AeRaHHI5NsSw,146
 arize/_generated/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arize/_generated/api_client_README.md,sha256=AM6a2RwMFfDPPt1nCFC-SYW0T4vNAwNfId9FYcPFjMM,4844
@@ -71,7 +71,7 @@ arize/experiments/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 arize/experiments/client.py,sha256=2fDq0fr_h6Knn_9zgDAlAhSUCKUrKozGLOQRTInCr4c,344
 arize/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arize/models/bounded_executor.py,sha256=o-PJsDAXQdiJ9dc-jzGCHMhT0-QBY9bvl4Ckn1017Eo,1131
-arize/models/client.py,sha256=zGk4NGOFskQn5tLOrN9OiAF0UmfE03PpSGez5qmt5ic,31395
+arize/models/client.py,sha256=aYXPv5Pq2Va2_aEEptw6-iD5zDEFV4UJz2bPnXvvIHw,31419
 arize/models/stream_validation.py,sha256=PtmqWgRdCxVtTNkHHEHIM1S6ECbYLA1vuQQFBw_t3Lw,7118
 arize/models/batch_validation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arize/models/batch_validation/errors.py,sha256=__I8l25zf4kGv6qgiwEm9LzGNgqmMSM8Fb88pBtyMxE,39990
@@ -86,7 +86,7 @@ arize/spans/validation/annotations/annotations_validation.py,sha256=fWoLFo98CPSz
 arize/spans/validation/annotations/dataframe_form_validation.py,sha256=wrYd81ZT5MxzbXqNUqq-m1NGC-rT7upoDqIBzycvfi8,7028
 arize/spans/validation/annotations/value_validation.py,sha256=Xk97D0ymV8KoMP8Zr9vqVq6Clwf8MMN1IfbcAt5W80s,7655
 arize/spans/validation/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arize/spans/validation/common/argument_validation.py,sha256=m5UVtYbX39-VRemJe2pv50RgaBoLjoD-HLFN-Tz6pM0,1542
+arize/spans/validation/common/argument_validation.py,sha256=VvR1xQLHcSOFeCAaNhAbYgtUIvbHd_NK5RlES0I7OZ8,1469
 arize/spans/validation/common/dataframe_form_validation.py,sha256=g9BUyIJYdEx2Uf-pkRxeaAdY-u1lWzYdxB5-N8yPoCE,1294
 arize/spans/validation/common/errors.py,sha256=iZTZKy-Me8BtOFIzNg9gjg7UqRRkBGPnFsC_ZhUeDNQ,13478
 arize/spans/validation/common/value_validation.py,sha256=s0mZAoHhg6iT5lNF5EcWn5Z0ERr_b-62GOIDT1y1awE,6968
@@ -95,7 +95,7 @@ arize/spans/validation/evals/dataframe_form_validation.py,sha256=LAcWQ23Fb3XbYTS
 arize/spans/validation/evals/evals_validation.py,sha256=GVFkgJ4VmvIfPuiLleQZhGjd-CHfXbjGwDjqon4Qhbc,2268
 arize/spans/validation/evals/value_validation.py,sha256=h7Rss51asFgwhCAVlJ4WHYgyt1DCT-FW_AtLqy4uMig,3305
 arize/spans/validation/metadata/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-arize/spans/validation/metadata/argument_validation.py,sha256=CHpPqMp7ipdwuv0O1tOIJmOu0PT6eAh_ws3TXvjq6Vk,1552
+arize/spans/validation/metadata/argument_validation.py,sha256=vsVbmg1ZC2DpbaWtMpU1fnEQy7ik1Vdjo8Au4THgqDE,1479
 arize/spans/validation/metadata/dataframe_form_validation.py,sha256=E5gXq9SmDXigUHMaTSCInWlsF-GZxIRQz-GvoH_zWRo,3441
 arize/spans/validation/metadata/value_validation.py,sha256=1DFshKQhy7jwA97tRlvwnnEsUbd1-tfmcWgG6cZO1uQ,9741
 arize/spans/validation/spans/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -103,11 +103,11 @@ arize/spans/validation/spans/dataframe_form_validation.py,sha256=NqyaOYzZL5_XIHR
 arize/spans/validation/spans/spans_validation.py,sha256=p6IjbQMtOhotGBfw3axj7yMWxb2pC47EAzJSB887nzs,2451
 arize/spans/validation/spans/value_validation.py,sha256=H3qV96w6JQNCed_MxhWDas9Jf6vUj6RFabShcwf4jr4,19102
 arize/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arize/utils/arrow.py,sha256=Tb4HQiiFBScYA3MtrhB4ppuHqRxToQ0-rkixZ59cVa0,5447
+arize/utils/arrow.py,sha256=4In1gQc0i4Rb8zuwI0w-Hv-10wiItu5opqqGrJ8tSzo,5277
 arize/utils/casting.py,sha256=KUrPUQN6qJEVe39nxbr0T-0GjAJLHjf4xWuzV71QezI,12468
 arize/utils/dataframe.py,sha256=I0FloPgNiqlKga32tMOvTE70598QA8Hhrgf-6zjYMAM,1120
 arize/utils/proto.py,sha256=9vLo53INYjdF78ffjm3E48jFwK6LbPD2FfKei7VaDy8,35477
-arize-8.0.0a7.dist-info/METADATA,sha256=sAC-xW3yGRagFYnCnnmjg5dKd5jb-JSlTbx0hNddF4E,12346
-arize-8.0.0a7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arize-8.0.0a7.dist-info/licenses/LICENSE.md,sha256=8vLN8Gms62NCBorxIv9MUvuK7myueb6_-dhXHPmm4H0,1479
-arize-8.0.0a7.dist-info/RECORD,,
+arize-8.0.0a9.dist-info/METADATA,sha256=WGsj9jvMjlNY9Xy4c9GpIT6mkJ1551ITCUYEhhQESck,12453
+arize-8.0.0a9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arize-8.0.0a9.dist-info/licenses/LICENSE.md,sha256=8vLN8Gms62NCBorxIv9MUvuK7myueb6_-dhXHPmm4H0,1479
+arize-8.0.0a9.dist-info/RECORD,,