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)

Commits related to this issue

Most upvoted comments

@alvarobartt maybe we can add a try/except block in rg.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.

Sure, we can do that! Do you want to pick that up @gabrielmbmb? 😄

@alvarobartt maybe we can add a try/except block in rg.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.