Skip to content

Commit 0434bc0

Browse files
Add RL modality (#90)
* Add RL modality * Update js/src/lib/interfaces/Types.ts Co-authored-by: Omar Sanseviero <[email protected]> Co-authored-by: Omar Sanseviero <[email protected]>
1 parent 00c247f commit 0434bc0

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

js/src/lib/interfaces/Types.ts

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ export const MODALITIES = [
6666
"nlp",
6767
"audio",
6868
"cv",
69+
"rl",
6970
"other",
7071
] as const;
7172

@@ -75,6 +76,7 @@ export const MODALITY_LABELS: Record<Modality, string> = {
7576
nlp: "Natural Language Processing",
7677
audio: "Audio",
7778
cv: "Computer Vision",
79+
rl: "Reinforcement Learning",
7880
other: "Other",
7981
};
8082

@@ -104,8 +106,8 @@ export const PIPELINE_TAG_MODALITIES: Record<keyof typeof PipelineType, Modality
104106
"image-to-text": "cv",
105107
"image-to-image": "cv",
106108
"unconditional-image-generation": "cv",
109+
"reinforcement-learning": "rl",
107110
"structured-data-classification": "other",
108-
"reinforcement-learning": "other",
109111
};
110112

111113
/*

0 commit comments

Comments
 (0)