Import the required classes and initialize Athina API key.

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

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

Delete Dataset By ID

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