Fix js/lite/src/custom-element/index.test.ts that has been failing frequently (#10287)

* Fix js/lite/src/custom-element/index.test.ts that has been failing frequently

* Delete the vite-environment specification

Author: whitphx

js/lite/src/custom-element/index.test.ts

@@ -1,8 +1,8 @@
 import { describe, it, expect, vi, beforeEach } from "vitest";
 import { bootstrap_custom_element } from "./index";
 
-const delay = (ms?: number) =>
-	new Promise((resolve) => setTimeout(resolve, ms));
+const awaitAnimationFrame = () =>
+	new Promise((resolve) => window.requestAnimationFrame(resolve));
 
 describe("bootstrap_custom_element", () => {
 	const create = vi.fn();
@@ -20,7 +20,7 @@ describe("bootstrap_custom_element", () => {
   import gradio as gr
 </gradio-lite>
 `;
-		await delay(); // Wait for the requestAnimationFrame to run
+		await awaitAnimationFrame(); // Wait for `create` to be called in the `requestAnimationFrame` callback
 
 		expect(create).toHaveBeenCalledWith(
 			expect.objectContaining({
@@ -44,7 +44,7 @@ scipy
     </gradio-requirements>
 </gradio-lite>
 `;
-		await delay(); // Wait for the requestAnimationFrame to run
+		await awaitAnimationFrame(); // Wait for `create` to be called in the `requestAnimationFrame` callback
 
 		expect(create).toHaveBeenCalledWith(
 			expect.objectContaining({
@@ -75,7 +75,7 @@ scipy
     </gradio-requirements>
 </gradio-lite>
 `;
-		await delay(); // Wait for the requestAnimationFrame to run
+		await awaitAnimationFrame(); // Wait for `create` to be called in the `requestAnimationFrame` callback
 
 		expect(create).toHaveBeenCalledWith(
 			expect.objectContaining({
@@ -100,7 +100,7 @@ scipy
 		import gradio as gr
 </gradio-lite>
 `;
-		await delay(); // Wait for the requestAnimationFrame to run
+		await awaitAnimationFrame(); // Wait for `create` to be called in the `requestAnimationFrame` callback
 
 		expect(create).toHaveBeenCalledWith(
 			expect.objectContaining({
@@ -119,7 +119,7 @@ scipy
 		</gradio-code>
 </gradio-lite>
 `;
-		await delay(); // Wait for the requestAnimationFrame to run
+		await awaitAnimationFrame(); // Wait for `create` to be called in the `requestAnimationFrame` callback
 
 		expect(create).toHaveBeenCalledWith(
 			expect.objectContaining({
@@ -149,7 +149,7 @@ scipy
 		</gradio-file>
 </gradio-lite>
 `;
-		await delay(); // Wait for the requestAnimationFrame to run
+		await awaitAnimationFrame(); // Wait for `create` to be called in the `requestAnimationFrame` callback
 
 		expect(create).toHaveBeenCalledWith(
 			expect.objectContaining({