@huggingface/tasks 0.0.3 → 0.0.5

package/dist/index.js

@@ -1,8 +1,46 @@
-// ../js/src/lib/interfaces/Types.ts
-function ensureRecordOfPipelines(record) {
-  return record;
-}
-var PIPELINE_DATA = ensureRecordOfPipelines({
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  MODALITIES: () => MODALITIES,
+  MODALITY_LABELS: () => MODALITY_LABELS,
+  ModelLibrary: () => ModelLibrary,
+  PIPELINE_DATA: () => PIPELINE_DATA,
+  PIPELINE_TYPES: () => PIPELINE_TYPES,
+  TASKS_DATA: () => TASKS_DATA
+});
+module.exports = __toCommonJS(src_exports);
+
+// src/pipelines.ts
+var MODALITIES = ["cv", "nlp", "audio", "tabular", "multimodal", "rl", "other"];
+var MODALITY_LABELS = {
+  multimodal: "Multimodal",
+  nlp: "Natural Language Processing",
+  audio: "Audio",
+  cv: "Computer Vision",
+  rl: "Reinforcement Learning",
+  tabular: "Tabular",
+  other: "Other"
+};
+var PIPELINE_DATA = {
   "text-classification": {
     name: "Text Classification",
     subtasks: [
@@ -138,12 +176,12 @@ var PIPELINE_DATA = ensureRecordOfPipelines({
     modality: "nlp",
     color: "yellow"
   },
-  "translation": {
+  translation: {
     name: "Translation",
     modality: "nlp",
     color: "green"
   },
-  "summarization": {
+  summarization: {
     name: "Summarization",
     subtasks: [
       {
@@ -158,7 +196,7 @@ var PIPELINE_DATA = ensureRecordOfPipelines({
     modality: "nlp",
     color: "indigo"
   },
-  "conversational": {
+  conversational: {
     name: "Conversational",
     subtasks: [
       {
@@ -386,7 +424,7 @@ var PIPELINE_DATA = ensureRecordOfPipelines({
     modality: "rl",
     color: "red"
   },
-  "robotics": {
+  robotics: {
     name: "Robotics",
     modality: "rl",
     subtasks: [
@@ -539,19 +577,15 @@ var PIPELINE_DATA = ensureRecordOfPipelines({
     modality: "multimodal",
     color: "green"
   },
-  "other": {
+  other: {
     name: "Other",
     modality: "other",
     color: "blue",
     hideInModels: true,
     hideInDatasets: true
   }
-});
-var ALL_PIPELINE_TYPES = Object.keys(PIPELINE_DATA);
-var ALL_PIPELINE_TYPES_SET = new Set(ALL_PIPELINE_TYPES);
-var ALL_SUBTASKS = Object.values(PIPELINE_DATA).flatMap((data) => data.subtasks ?? []);
-var ALL_SUBTASK_TYPES = ALL_SUBTASKS.map((s) => s.type);
-var ALL_SUBTASK_TYPES_SET = new Set(ALL_SUBTASK_TYPES);
+};
+var PIPELINE_TYPES = Object.keys(PIPELINE_DATA);
 
 // src/audio-classification/data.ts
 var taskData = {
@@ -1839,19 +1873,9 @@ var taskData19 = {
         ],
         type: "tabular"
       },
-      {
-        label: "Question",
-        content: "What is the number of reigns for Harley Race?",
-        type: "text"
-      }
+      { label: "Question", content: "What is the number of reigns for Harley Race?", type: "text" }
     ],
-    outputs: [
-      {
-        label: "Result",
-        content: "7",
-        type: "text"
-      }
-    ]
+    outputs: [{ label: "Result", content: "7", type: "text" }]
   },
   metrics: [
     {
@@ -2978,7 +3002,7 @@ var TASKS_MODEL_LIBRARIES = {
   "audio-classification": ["speechbrain", "transformers"],
   "audio-to-audio": ["asteroid", "speechbrain"],
   "automatic-speech-recognition": ["espnet", "nemo", "speechbrain", "transformers", "transformers.js"],
-  "conversational": ["transformers"],
+  conversational: ["transformers"],
   "depth-estimation": ["transformers"],
   "document-question-answering": ["transformers"],
   "feature-extraction": ["sentence-transformers", "transformers", "transformers.js"],
@@ -2991,12 +3015,12 @@ var TASKS_MODEL_LIBRARIES = {
   "video-classification": [],
   "multiple-choice": ["transformers"],
   "object-detection": ["transformers", "transformers.js"],
-  "other": [],
+  other: [],
   "question-answering": ["adapter-transformers", "allennlp", "transformers", "transformers.js"],
-  "robotics": [],
+  robotics: [],
   "reinforcement-learning": ["transformers", "stable-baselines3", "ml-agents", "sample-factory"],
   "sentence-similarity": ["sentence-transformers", "spacy", "transformers.js"],
-  "summarization": ["transformers", "transformers.js"],
+  summarization: ["transformers", "transformers.js"],
   "table-question-answering": ["transformers"],
   "table-to-text": ["transformers"],
   "tabular-classification": ["sklearn"],
@@ -3011,8 +3035,16 @@ var TASKS_MODEL_LIBRARIES = {
   "text-to-video": [],
   "text2text-generation": ["transformers", "transformers.js"],
   "time-series-forecasting": [],
-  "token-classification": ["adapter-transformers", "flair", "spacy", "span-marker", "stanza", "transformers", "transformers.js"],
-  "translation": ["transformers", "transformers.js"],
+  "token-classification": [
+    "adapter-transformers",
+    "flair",
+    "spacy",
+    "span-marker",
+    "stanza",
+    "transformers",
+    "transformers.js"
+  ],
+  translation: ["transformers", "transformers.js"],
   "unconditional-image-generation": [],
   "visual-question-answering": [],
   "voice-activity-detection": [],
@@ -3025,7 +3057,7 @@ var TASKS_DATA = {
   "audio-classification": getData("audio-classification", data_default),
   "audio-to-audio": getData("audio-to-audio", data_default2),
   "automatic-speech-recognition": getData("automatic-speech-recognition", data_default3),
-  "conversational": getData("conversational", data_default4),
+  conversational: getData("conversational", data_default4),
   "depth-estimation": getData("depth-estimation", data_default13),
   "document-question-answering": getData("document-question-answering", data_default5),
   "feature-extraction": getData("feature-extraction", data_default6),
@@ -3038,12 +3070,12 @@ var TASKS_DATA = {
   "multiple-choice": void 0,
   "object-detection": getData("object-detection", data_default12),
   "video-classification": getData("video-classification", data_default30),
-  "other": void 0,
+  other: void 0,
   "question-answering": getData("question-answering", data_default16),
   "reinforcement-learning": getData("reinforcement-learning", data_default15),
-  "robotics": void 0,
+  robotics: void 0,
   "sentence-similarity": getData("sentence-similarity", data_default17),
-  "summarization": getData("summarization", data_default18),
+  summarization: getData("summarization", data_default18),
   "table-question-answering": getData("table-question-answering", data_default19),
   "table-to-text": void 0,
   "tabular-classification": getData("tabular-classification", data_default20),
@@ -3059,7 +3091,7 @@ var TASKS_DATA = {
   "text2text-generation": void 0,
   "time-series-forecasting": void 0,
   "token-classification": getData("token-classification", data_default24),
-  "translation": getData("translation", data_default25),
+  translation: getData("translation", data_default25),
   "unconditional-image-generation": getData("unconditional-image-generation", data_default29),
   "visual-question-answering": getData("visual-question-answering", data_default31),
   "voice-activity-detection": void 0,
@@ -3074,6 +3106,50 @@ function getData(type, partialTaskData = data_default14) {
     libraries: TASKS_MODEL_LIBRARIES[type]
   };
 }
-export {
+
+// src/modelLibraries.ts
+var ModelLibrary = /* @__PURE__ */ ((ModelLibrary2) => {
+  ModelLibrary2["adapter-transformers"] = "Adapter Transformers";
+  ModelLibrary2["allennlp"] = "allenNLP";
+  ModelLibrary2["asteroid"] = "Asteroid";
+  ModelLibrary2["bertopic"] = "BERTopic";
+  ModelLibrary2["diffusers"] = "Diffusers";
+  ModelLibrary2["doctr"] = "docTR";
+  ModelLibrary2["espnet"] = "ESPnet";
+  ModelLibrary2["fairseq"] = "Fairseq";
+  ModelLibrary2["flair"] = "Flair";
+  ModelLibrary2["keras"] = "Keras";
+  ModelLibrary2["k2"] = "K2";
+  ModelLibrary2["nemo"] = "NeMo";
+  ModelLibrary2["open_clip"] = "OpenCLIP";
+  ModelLibrary2["paddlenlp"] = "PaddleNLP";
+  ModelLibrary2["peft"] = "PEFT";
+  ModelLibrary2["pyannote-audio"] = "pyannote.audio";
+  ModelLibrary2["sample-factory"] = "Sample Factory";
+  ModelLibrary2["sentence-transformers"] = "Sentence Transformers";
+  ModelLibrary2["sklearn"] = "Scikit-learn";
+  ModelLibrary2["spacy"] = "spaCy";
+  ModelLibrary2["span-marker"] = "SpanMarker";
+  ModelLibrary2["speechbrain"] = "speechbrain";
+  ModelLibrary2["tensorflowtts"] = "TensorFlowTTS";
+  ModelLibrary2["timm"] = "Timm";
+  ModelLibrary2["fastai"] = "fastai";
+  ModelLibrary2["transformers"] = "Transformers";
+  ModelLibrary2["transformers.js"] = "Transformers.js";
+  ModelLibrary2["stanza"] = "Stanza";
+  ModelLibrary2["fasttext"] = "fastText";
+  ModelLibrary2["stable-baselines3"] = "Stable-Baselines3";
+  ModelLibrary2["ml-agents"] = "ML-Agents";
+  ModelLibrary2["pythae"] = "Pythae";
+  ModelLibrary2["mindspore"] = "MindSpore";
+  return ModelLibrary2;
+})(ModelLibrary || {});
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  MODALITIES,
+  MODALITY_LABELS,
+  ModelLibrary,
+  PIPELINE_DATA,
+  PIPELINE_TYPES,
   TASKS_DATA
-};
+});

package/dist/{index.cjs → index.mjs}

@@ -1,34 +1,15 @@
-"use strict";
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
+// src/pipelines.ts
+var MODALITIES = ["cv", "nlp", "audio", "tabular", "multimodal", "rl", "other"];
+var MODALITY_LABELS = {
+  multimodal: "Multimodal",
+  nlp: "Natural Language Processing",
+  audio: "Audio",
+  cv: "Computer Vision",
+  rl: "Reinforcement Learning",
+  tabular: "Tabular",
+  other: "Other"
 };
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-
-// src/index.ts
-var src_exports = {};
-__export(src_exports, {
-  TASKS_DATA: () => TASKS_DATA
-});
-module.exports = __toCommonJS(src_exports);
-
-// ../js/src/lib/interfaces/Types.ts
-function ensureRecordOfPipelines(record) {
-  return record;
-}
-var PIPELINE_DATA = ensureRecordOfPipelines({
+var PIPELINE_DATA = {
   "text-classification": {
     name: "Text Classification",
     subtasks: [
@@ -164,12 +145,12 @@ var PIPELINE_DATA = ensureRecordOfPipelines({
     modality: "nlp",
     color: "yellow"
   },
-  "translation": {
+  translation: {
     name: "Translation",
     modality: "nlp",
     color: "green"
   },
-  "summarization": {
+  summarization: {
     name: "Summarization",
     subtasks: [
       {
@@ -184,7 +165,7 @@ var PIPELINE_DATA = ensureRecordOfPipelines({
     modality: "nlp",
     color: "indigo"
   },
-  "conversational": {
+  conversational: {
     name: "Conversational",
     subtasks: [
       {
@@ -412,7 +393,7 @@ var PIPELINE_DATA = ensureRecordOfPipelines({
     modality: "rl",
     color: "red"
   },
-  "robotics": {
+  robotics: {
     name: "Robotics",
     modality: "rl",
     subtasks: [
@@ -565,19 +546,15 @@ var PIPELINE_DATA = ensureRecordOfPipelines({
     modality: "multimodal",
     color: "green"
   },
-  "other": {
+  other: {
     name: "Other",
     modality: "other",
     color: "blue",
     hideInModels: true,
     hideInDatasets: true
   }
-});
-var ALL_PIPELINE_TYPES = Object.keys(PIPELINE_DATA);
-var ALL_PIPELINE_TYPES_SET = new Set(ALL_PIPELINE_TYPES);
-var ALL_SUBTASKS = Object.values(PIPELINE_DATA).flatMap((data) => data.subtasks ?? []);
-var ALL_SUBTASK_TYPES = ALL_SUBTASKS.map((s) => s.type);
-var ALL_SUBTASK_TYPES_SET = new Set(ALL_SUBTASK_TYPES);
+};
+var PIPELINE_TYPES = Object.keys(PIPELINE_DATA);
 
 // src/audio-classification/data.ts
 var taskData = {
@@ -1865,19 +1842,9 @@ var taskData19 = {
         ],
         type: "tabular"
       },
-      {
-        label: "Question",
-        content: "What is the number of reigns for Harley Race?",
-        type: "text"
-      }
+      { label: "Question", content: "What is the number of reigns for Harley Race?", type: "text" }
     ],
-    outputs: [
-      {
-        label: "Result",
-        content: "7",
-        type: "text"
-      }
-    ]
+    outputs: [{ label: "Result", content: "7", type: "text" }]
   },
   metrics: [
     {
@@ -3004,7 +2971,7 @@ var TASKS_MODEL_LIBRARIES = {
   "audio-classification": ["speechbrain", "transformers"],
   "audio-to-audio": ["asteroid", "speechbrain"],
   "automatic-speech-recognition": ["espnet", "nemo", "speechbrain", "transformers", "transformers.js"],
-  "conversational": ["transformers"],
+  conversational: ["transformers"],
   "depth-estimation": ["transformers"],
   "document-question-answering": ["transformers"],
   "feature-extraction": ["sentence-transformers", "transformers", "transformers.js"],
@@ -3017,12 +2984,12 @@ var TASKS_MODEL_LIBRARIES = {
   "video-classification": [],
   "multiple-choice": ["transformers"],
   "object-detection": ["transformers", "transformers.js"],
-  "other": [],
+  other: [],
   "question-answering": ["adapter-transformers", "allennlp", "transformers", "transformers.js"],
-  "robotics": [],
+  robotics: [],
   "reinforcement-learning": ["transformers", "stable-baselines3", "ml-agents", "sample-factory"],
   "sentence-similarity": ["sentence-transformers", "spacy", "transformers.js"],
-  "summarization": ["transformers", "transformers.js"],
+  summarization: ["transformers", "transformers.js"],
   "table-question-answering": ["transformers"],
   "table-to-text": ["transformers"],
   "tabular-classification": ["sklearn"],
@@ -3037,8 +3004,16 @@ var TASKS_MODEL_LIBRARIES = {
   "text-to-video": [],
   "text2text-generation": ["transformers", "transformers.js"],
   "time-series-forecasting": [],
-  "token-classification": ["adapter-transformers", "flair", "spacy", "span-marker", "stanza", "transformers", "transformers.js"],
-  "translation": ["transformers", "transformers.js"],
+  "token-classification": [
+    "adapter-transformers",
+    "flair",
+    "spacy",
+    "span-marker",
+    "stanza",
+    "transformers",
+    "transformers.js"
+  ],
+  translation: ["transformers", "transformers.js"],
   "unconditional-image-generation": [],
   "visual-question-answering": [],
   "voice-activity-detection": [],
@@ -3051,7 +3026,7 @@ var TASKS_DATA = {
   "audio-classification": getData("audio-classification", data_default),
   "audio-to-audio": getData("audio-to-audio", data_default2),
   "automatic-speech-recognition": getData("automatic-speech-recognition", data_default3),
-  "conversational": getData("conversational", data_default4),
+  conversational: getData("conversational", data_default4),
   "depth-estimation": getData("depth-estimation", data_default13),
   "document-question-answering": getData("document-question-answering", data_default5),
   "feature-extraction": getData("feature-extraction", data_default6),
@@ -3064,12 +3039,12 @@ var TASKS_DATA = {
   "multiple-choice": void 0,
   "object-detection": getData("object-detection", data_default12),
   "video-classification": getData("video-classification", data_default30),
-  "other": void 0,
+  other: void 0,
   "question-answering": getData("question-answering", data_default16),
   "reinforcement-learning": getData("reinforcement-learning", data_default15),
-  "robotics": void 0,
+  robotics: void 0,
   "sentence-similarity": getData("sentence-similarity", data_default17),
-  "summarization": getData("summarization", data_default18),
+  summarization: getData("summarization", data_default18),
   "table-question-answering": getData("table-question-answering", data_default19),
   "table-to-text": void 0,
   "tabular-classification": getData("tabular-classification", data_default20),
@@ -3085,7 +3060,7 @@ var TASKS_DATA = {
   "text2text-generation": void 0,
   "time-series-forecasting": void 0,
   "token-classification": getData("token-classification", data_default24),
-  "translation": getData("translation", data_default25),
+  translation: getData("translation", data_default25),
   "unconditional-image-generation": getData("unconditional-image-generation", data_default29),
   "visual-question-answering": getData("visual-question-answering", data_default31),
   "voice-activity-detection": void 0,
@@ -3100,7 +3075,49 @@ function getData(type, partialTaskData = data_default14) {
     libraries: TASKS_MODEL_LIBRARIES[type]
   };
 }
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
+
+// src/modelLibraries.ts
+var ModelLibrary = /* @__PURE__ */ ((ModelLibrary2) => {
+  ModelLibrary2["adapter-transformers"] = "Adapter Transformers";
+  ModelLibrary2["allennlp"] = "allenNLP";
+  ModelLibrary2["asteroid"] = "Asteroid";
+  ModelLibrary2["bertopic"] = "BERTopic";
+  ModelLibrary2["diffusers"] = "Diffusers";
+  ModelLibrary2["doctr"] = "docTR";
+  ModelLibrary2["espnet"] = "ESPnet";
+  ModelLibrary2["fairseq"] = "Fairseq";
+  ModelLibrary2["flair"] = "Flair";
+  ModelLibrary2["keras"] = "Keras";
+  ModelLibrary2["k2"] = "K2";
+  ModelLibrary2["nemo"] = "NeMo";
+  ModelLibrary2["open_clip"] = "OpenCLIP";
+  ModelLibrary2["paddlenlp"] = "PaddleNLP";
+  ModelLibrary2["peft"] = "PEFT";
+  ModelLibrary2["pyannote-audio"] = "pyannote.audio";
+  ModelLibrary2["sample-factory"] = "Sample Factory";
+  ModelLibrary2["sentence-transformers"] = "Sentence Transformers";
+  ModelLibrary2["sklearn"] = "Scikit-learn";
+  ModelLibrary2["spacy"] = "spaCy";
+  ModelLibrary2["span-marker"] = "SpanMarker";
+  ModelLibrary2["speechbrain"] = "speechbrain";
+  ModelLibrary2["tensorflowtts"] = "TensorFlowTTS";
+  ModelLibrary2["timm"] = "Timm";
+  ModelLibrary2["fastai"] = "fastai";
+  ModelLibrary2["transformers"] = "Transformers";
+  ModelLibrary2["transformers.js"] = "Transformers.js";
+  ModelLibrary2["stanza"] = "Stanza";
+  ModelLibrary2["fasttext"] = "fastText";
+  ModelLibrary2["stable-baselines3"] = "Stable-Baselines3";
+  ModelLibrary2["ml-agents"] = "ML-Agents";
+  ModelLibrary2["pythae"] = "Pythae";
+  ModelLibrary2["mindspore"] = "MindSpore";
+  return ModelLibrary2;
+})(ModelLibrary || {});
+export {
+  MODALITIES,
+  MODALITY_LABELS,
+  ModelLibrary,
+  PIPELINE_DATA,
+  PIPELINE_TYPES,
   TASKS_DATA
-});
+};

package/package.json

@@ -1,34 +1,44 @@
 {
-	"name": "@huggingface/tasks",
-	"version": "0.0.3",
-	"type": "module",
-	"description": "A library of tasks for the Hugging Face Hub",
-	"main": "./dist/index.cjs",
-	"module": "./dist/index.js",
-	"types": "./dist/index.d.ts",
-	"exports": {
-		".": {
-			"types": "./dist/index.d.ts",
-			"require": "./dist/index.cjs",
-			"import": "./dist/index.js"
-		}
-	},
-	"scripts": {
-		"prepublishOnly": "npm run build",
-		"build": "tsup src/index.ts --format cjs,esm --clean --dts",
-		"type-check": "tsc"
-	},
-	"source": "src/index.ts",
-	"files": [
-		"assets",
-		"src",
-		"dist"
-	],
-	"devDependencies": {
-		"tsup": "^7.3.0",
-		"typescript": "^5.2.2"
-	},
-	"publishConfig": {
-		"access": "public"
-	}
-}
+  "name": "@huggingface/tasks",
+  "packageManager": "pnpm@8.10.5",
+  "version": "0.0.5",
+  "description": "List of ML tasks for huggingface.co/tasks",
+  "repository": "https://github.com/huggingface/huggingface.js.git",
+  "publishConfig": {
+    "access": "public"
+  },
+  "main": "./dist/index.js",
+  "module": "./dist/index.mjs",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "require": "./dist/index.js",
+      "import": "./dist/index.mjs"
+    }
+  },
+  "source": "src/index.ts",
+  "files": [
+    "dist",
+    "src",
+    "tsconfig.json"
+  ],
+  "keywords": [
+    "huggingface",
+    "hub",
+    "languages"
+  ],
+  "author": "Hugging Face",
+  "license": "MIT",
+  "devDependencies": {
+    "typescript": "^5.0.4"
+  },
+  "scripts": {
+    "lint": "eslint --quiet --fix --ext .cjs,.ts .",
+    "lint:check": "eslint --ext .cjs,.ts .",
+    "format": "prettier --write .",
+    "format:check": "prettier --check .",
+    "build": "tsup src/index.ts --format cjs,esm --clean --dts",
+    "check": "tsc"
+  }
+}