From ef74f5edb10482e318a624c8d468d2ff5a6b1df9 Mon Sep 17 00:00:00 2001
From: yuneng-jiang <yuneng.jiang@gmail.com>
Date: Sat, 10 Jan 2026 15:15:27 -0800
Subject: [PATCH] fixing test

---
 .../EndpointUsageLineChart.test.tsx           | 26 -------------------
 .../components/EndpointUsageLineChart.tsx     |  2 +-
 2 files changed, 1 insertion(+), 27 deletions(-)

diff --git a/ui/litellm-dashboard/src/components/UsagePage/components/EndpointUsage/components/EndpointUsageLineChart.test.tsx b/ui/litellm-dashboard/src/components/UsagePage/components/EndpointUsage/components/EndpointUsageLineChart.test.tsx
index ec12779a69..ec99825289 100644
--- a/ui/litellm-dashboard/src/components/UsagePage/components/EndpointUsage/components/EndpointUsageLineChart.test.tsx
+++ b/ui/litellm-dashboard/src/components/UsagePage/components/EndpointUsage/components/EndpointUsageLineChart.test.tsx
@@ -23,38 +23,12 @@ vi.mock("@tremor/react", async () => {
   return { Card, Title, LineChart };
 });
 
-vi.mock("antd", async () => {
-  const React = await import("react");
-
-  function Segmented({ options, value, onChange }: any) {
-    return React.createElement(
-      "div",
-      { "data-testid": "antd-segmented", "data-value": value },
-      options?.map((opt: any) =>
-        React.createElement(
-          "button",
-          {
-            key: opt.value,
-            onClick: () => onChange?.(opt.value),
-            "data-selected": value === opt.value,
-          },
-          opt.label,
-        ),
-      ),
-    );
-  }
-  (Segmented as any).displayName = "Segmented";
-
-  return { Segmented };
-});
-
 describe("EndpointUsageLineChart", () => {
   it("should render", () => {
     render(<EndpointUsageLineChart />);
 
     expect(screen.getByTestId("tremor-card")).toBeInTheDocument();
     expect(screen.getByText("Endpoint Usage Trends")).toBeInTheDocument();
-    expect(screen.getByTestId("antd-segmented")).toBeInTheDocument();
     expect(screen.getByTestId("tremor-line-chart")).toBeInTheDocument();
   });
 });
diff --git a/ui/litellm-dashboard/src/components/UsagePage/components/EndpointUsage/components/EndpointUsageLineChart.tsx b/ui/litellm-dashboard/src/components/UsagePage/components/EndpointUsage/components/EndpointUsageLineChart.tsx
index dd113423e4..564eaa1fe8 100644
--- a/ui/litellm-dashboard/src/components/UsagePage/components/EndpointUsage/components/EndpointUsageLineChart.tsx
+++ b/ui/litellm-dashboard/src/components/UsagePage/components/EndpointUsage/components/EndpointUsageLineChart.tsx
@@ -1,5 +1,5 @@
-import React, { useMemo } from "react";
 import { Card, LineChart, Title } from "@tremor/react";
+import { useMemo } from "react";
 import { DailyData } from "../../../types";
 
 interface EndpointUsageLineChartProps {