Get Dataset

import os
from athina_client.datasets import Dataset
from athina_client.keys import AthinaApiKey

AthinaApiKey.set_key(os.getenv('ATHINA_API_KEY'))

# Get a dataset by ID
try:
    dataset = Dataset.get_dataset_by_id(dataset_id=DATASET_ID)
except Exception as e:
    print(f"Failed to get dataset: {e}")

# Get a dataset by Name
try:
    dataset = Dataset.get_dataset_by_name(name="test_dataset")
except Exception as e:
    print(f"Failed to get dataset: {e}")

# Get a dataset with annotations
try:
    # Using dataset ID
    dataset = Dataset.get_dataset_by_id(dataset_id=DATASET_ID, include_dataset_annotations=True)

    # Or using dataset name
    # dataset = Dataset.get_dataset_by_name(name="test_dataset", include_dataset_annotations=True)
except Exception as e:
    print(f"Failed to get dataset: {e}")