argilla: [BUG] Issue loading FeedbackDataset
Describe the bug After creating a feedback data set when I try to use rg.load(name = feedbackDatasetName) I get an error.
Stacktrace and Code to create the bug
questions = [
rg.RatingQuestion(
name="response_ranking",
title="Rank the responses\n1: first response is better,\n 2: second response is better,\n 3: both are equal ",
required=True,
values=[1, 2, 3]
),
rg.TextQuestion(
name="correct_response",
title="If none of the responses are helpful and correct, provide the response",
required=False
),
]
fields = [
rg.TextField(name="prompt", required=True),
rg.TextField(name="response-1", required=True),
rg.TextField(name="response-2", required=True)
]
dataset = rg.FeedbackDataset(
guidelines="Please, read the prompt carefully and...",
questions=questions,
fields=fields
)
record = rg.FeedbackRecord(fields={"prompt": "test", "response-1": "response 1", "response-2": "response_2"})
records.append(record)
# Add records to the dataset
dataset.add_records(records)
# This publishes the dataset and pushes the records into Argilla
dataset.push_to_argilla(name="feedback_testing ")
# load predictions for review
dataset_rg = rg.load(
name="feedback_testing"
)
Error is:
NotFoundApiError: Argilla server returned an error with http status: 404 Error details: [{‘code’: ‘argilla.api.errors::EntityNotFoundError’, ‘params’: {‘name’: ‘safety_testing’, ‘type’: ‘ServiceDataset’}}]
Expected behavior I expect to read a dataset like I do any other dataset
Environment:
- Argilla Version 1.9.0
- Argilla server 1.9.0
About this issue
- Original URL
- State: closed
- Created a year ago
- Comments: 17 (15 by maintainers)
Sure, we can do that! Do you want to pick that up @gabrielmbmb? 😄
@alvarobartt maybe we can add a
try/except
block inrg.load
and do an extra call to the API to check if there is a Feedback Dataset with the provided name and warn the user.