From b6b21c81c06e95d6b657c98165bae06d5549fabf Mon Sep 17 00:00:00 2001
From: Julien Chaumond <julien@huggingface.co>
Date: Fri, 29 Sep 2023 13:54:37 +0200
Subject: [PATCH] =?UTF-8?q?=F0=9F=A4=AF=20Use=20generics=20in=20Svelte=20(?=
 =?UTF-8?q?i=20guess...)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../WidgetInputSamples/WidgetInputSamples.svelte       | 10 +++++-----
 .../shared/WidgetWrapper/WidgetWrapper.svelte          |  6 +++---
 2 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/js/src/lib/components/InferenceWidget/shared/WidgetInputSamples/WidgetInputSamples.svelte b/js/src/lib/components/InferenceWidget/shared/WidgetInputSamples/WidgetInputSamples.svelte
index 3eaf7e278..59985626f 100644
--- a/js/src/lib/components/InferenceWidget/shared/WidgetInputSamples/WidgetInputSamples.svelte
+++ b/js/src/lib/components/InferenceWidget/shared/WidgetInputSamples/WidgetInputSamples.svelte
@@ -1,4 +1,4 @@
-<script lang="ts">
+<script lang="ts" generics="T extends WidgetExample">
 	import type { WidgetExample } from "../WidgetExample";
 
 	import { slide } from "svelte/transition";
@@ -8,8 +8,8 @@
 	export let classNames = "";
 	export let isLoading = false;
 	export let inputSamples: WidgetExample[];
-	export let applyInputSample: (sample: WidgetExample) => void;
-	export let previewInputSample: (sample: WidgetExample) => void;
+	export let applyInputSample: (sample: T) => void;
+	export let previewInputSample: (sample: T) => void;
 
 	let containerEl: HTMLElement;
 	let isOptionsVisible = false;
@@ -25,12 +25,12 @@
 		hideOptions();
 		const sample = inputSamples[idx];
 		title = sample.example_title as string;
-		applyInputSample(sample);
+		applyInputSample(sample as T);
 	}
 
 	function _previewInputSample(idx: number) {
 		const sample = inputSamples[idx];
-		previewInputSample(sample);
+		previewInputSample(sample as T);
 	}
 
 	function toggleOptionsVisibility() {
diff --git a/js/src/lib/components/InferenceWidget/shared/WidgetWrapper/WidgetWrapper.svelte b/js/src/lib/components/InferenceWidget/shared/WidgetWrapper/WidgetWrapper.svelte
index c7d356d68..4a4de0d18 100644
--- a/js/src/lib/components/InferenceWidget/shared/WidgetWrapper/WidgetWrapper.svelte
+++ b/js/src/lib/components/InferenceWidget/shared/WidgetWrapper/WidgetWrapper.svelte
@@ -1,4 +1,4 @@
-<script lang="ts">
+<script lang="ts" generics="T extends WidgetExample">
 	import type { WidgetProps, ModelLoadInfo } from "../types";
 	import type { WidgetExample } from "../WidgetExample";
 
@@ -26,8 +26,8 @@
 	};
 	export let noTitle = false;
 	export let outputJson: string;
-	export let applyInputSample: (sample: WidgetExample) => void = () => {};
-	export let previewInputSample: (sample: WidgetExample) => void = () => {};
+	export let applyInputSample: (sample: T) => void = () => {};
+	export let previewInputSample: (sample: T) => void = () => {};
 
 	let isMaximized = false;
 	let modelLoadInfo: ModelLoadInfo | undefined = undefined;