Troubleshooting Custom Attributes in ArizeExportClient Querying
hi i am having trouble with querying custom attributes using ArizeExportClient.export_model_to_df. the error is saying my attribute can't be found, but i can see it in the trace. i am using my custom attribute in both the where arg and columns arg
File ~/.local/share/virtualenvs/zenportai_mvp-ZWkb7xjr/lib/python3.13/site-packages/arize/exporter/core/client.py:132, in ArizeExportClient.export_model_to_df(self, space_id, model_id, environment, start_time, end_time, include_actuals, model_version, batch_id, where, similarity_search_params, columns, stream_chunk_size)
74 def export_model_to_df(
75 self,
76 space_id: str,
(...) 87 stream_chunk_size: Optional[int] = None,
88 ) -> pd.DataFrame:
89 """
90 Exports data of a specific model in the Arize platform to a pandas dataframe for a defined
91 time interval and model environment, optionally by model version and/or batch id.
(...) 130
131 """
--> 132 stream_reader, num_recs = self._get_model_stream_reader(
133 space_id=space_id,
134 model_id=model_id,
135 environment=environment,
136 start_time=start_time,
137 end_time=end_time,
138 include_actuals=include_actuals,
139 model_version=model_version,
140 batch_id=batch_id,
141 where=where,
142 similarity_search_params=similarity_search_params,
143 columns=columns,
144 stream_chunk_size=stream_chunk_size,
145 )
146 if stream_reader is None:
147 return pd.DataFrame()
File ~/.local/share/virtualenvs/zenportai_mvp-ZWkb7xjr/lib/python3.13/site-packages/arize/exporter/core/client.py:330, in ArizeExportClient._get_model_stream_reader(self, space_id, model_id, environment, start_time, end_time, include_actuals, model_version, batch_id, where, similarity_search_params, columns, stream_chunk_size)
328 flight_client = self.session.connect()
329 query = Query(query_descriptor)
--> 330 reader = query.execute(flight_client, self.session.call_options)
331 return reader
File ~/.local/share/virtualenvs/zenportai_mvp-ZWkb7xjr/lib/python3.13/site-packages/arize/exporter/core/query.py:22, in Query.execute(self, client, call_options)
16 def execute(
17 self,
18 client: flight.FlightClient,
19 call_options: flight.FlightCallOptions,
20 ) -> Tuple[flight.FlightStreamReader, int]:
21 try:
---> 22 flight_info = client.get_flight_info(
23 flight.FlightDescriptor.for_command(
24 json_format.MessageToJson(self.query_descriptor) # type: ignore
25 ),
26 call_options,
27 )
28 logger.info("Fetching data...")
30 if flight_info.total_records == 0:
File ~/.local/share/virtualenvs/zenportai_mvp-ZWkb7xjr/lib/python3.13/site-packages/pyarrow/_flight.pyx:1668, in pyarrow._flight.FlightClient.get_flight_info()
File ~/.local/share/virtualenvs/zenportai_mvp-ZWkb7xjr/lib/python3.13/site-packages/pyarrow/_flight.pyx:78, in pyarrow._flight.check_flight_status()
File ~/.local/share/virtualenvs/zenportai_mvp-ZWkb7xjr/lib/python3.13/site-packages/pyarrow/error.pxi:92, in pyarrow.lib.check_status()
ArrowInvalid: Flight returned invalid argument error, with message: Custom Filter Error: column "attributes.side_output.resolution" does not exist