Hi Shubham A., you can get your annotations by pulling down your trace dataset in AX & your annotations should be in that dataset by default. Here's a code snippet that might help:
from arize.exporter import ArizeExportClient
from arize.utils.types import Environments
from datetime import datetime, timedelta, timezone
client = ArizeExportClient(api_key=os.environ["ARIZE_API_KEY"])
model_id = "[insert project name]"
primary_df = client.export_model_to_df(
space_id=os.environ["ARIZE_SPACE_ID"],
model_id=model_id,
environment=Environments.TRACING,
start_time=datetime.now(timezone.utc) - timedelta(days=365),
end_time=datetime.now(timezone.utc)
)
primary_df['annotation.test.label']
Hope this helps for now! Please feel free to ask any more questions you have
