import argilla as rg

rg.init(
    api_url='https://brancengregory-demo-argilla.hf.space',
    api_key='team.apikey'
)

dataset = rg.load("plaintiff_sample").prepare_for_training()