From f153feb075cce26dac560678c86cb6f5be6674db Mon Sep 17 00:00:00 2001 From: Sylvain Lesage Date: Mon, 15 Jul 2024 16:19:47 +0200 Subject: [PATCH] add dataset libraries: distilabel, fiftyone, argilla (#797) Co-authored-by: burtenshaw --- packages/tasks/README.md | 2 +- packages/tasks/src/dataset-libraries.ts | 18 ++++++++++++++++++ 2 files changed, 19 insertions(+), 1 deletion(-) diff --git a/packages/tasks/README.md b/packages/tasks/README.md index 97132dbb2..3d49eb3a8 100644 --- a/packages/tasks/README.md +++ b/packages/tasks/README.md @@ -4,7 +4,7 @@ This package contains the definition files (written in Typescript) for the huggi - **pipeline types** (a.k.a. **task types**) - used to determine which widget to display on the model page, and which inference API to run. - **default widget inputs** - when they aren't provided in the model card. -- definitions and UI elements for **model libraries** (and soon for **dataset libraries**). +- definitions and UI elements for **model and dataset libraries**. Please add any missing ones to these definitions by opening a PR. Thanks 🔥 diff --git a/packages/tasks/src/dataset-libraries.ts b/packages/tasks/src/dataset-libraries.ts index 5f3f3e90d..854a78fd4 100644 --- a/packages/tasks/src/dataset-libraries.ts +++ b/packages/tasks/src/dataset-libraries.ts @@ -53,6 +53,24 @@ export const DATASET_LIBRARIES_UI_ELEMENTS = { repoUrl: "https://github.com/dask/dask", docsUrl: "https://huggingface.co/docs/hub/datasets-dask", }, + distilabel: { + prettyLabel: "Distilabel", + repoName: "distilabel", + repoUrl: "https://github.com/argilla-io/distilabel", + docsUrl: "https://distilabel.argilla.io", + }, + fiftyone: { + prettyLabel: "FiftyOne", + repoName: "fiftyone", + repoUrl: "https://github.com/voxel51/fiftyone", + docsUrl: "https://docs.voxel51.com", + }, + argilla: { + prettyLabel: "Argilla", + repoName: "argilla", + repoUrl: "https://github.com/argilla-io/argilla", + docsUrl: "https://argilla-io.github.io/argilla", + }, } satisfies Record; /// List of the dataset libraries supported by the Hub