Fix uneeded assume + assume after invite accept

Signed-off-by: Émile Ré <emile@getprobo.com>

packages/ui/src/Atoms/Badge/Badge.tsx

@@ -3,43 +3,45 @@ import { tv } from "tailwind-variants";
 import { Slot } from "../Slot";
 
 type Props = {
-    asChild?: boolean;
-    variant?:
-        | "success"
-        | "warning"
-        | "danger"
-        | "info"
-        | "neutral"
-        | "outline"
-        | "highlight";
-    size?: "sm" | "md";
+  asChild?: boolean;
+  variant?:
+    | "success"
+    | "warning"
+    | "danger"
+    | "info"
+    | "neutral"
+    | "outline"
+    | "highlight";
+  size?: "sm" | "md";
 } & HTMLAttributes<HTMLElement>;
 
 const badge = tv({
-    base: "font-medium rounded-lg w-max flex gap-1 items-center group whitespace-nowrap",
-    variants: {
-        variant: {
-            success: "bg-success text-txt-success",
-            warning: "bg-warning text-txt-warning",
-            danger: "bg-danger text-txt-danger",
-            info: "bg-info text-txt-info",
-            neutral: "bg-subtle text-txt-secondary",
-            outline: "text-txt-tertiary border border-border-low",
-            highlight: "bg-highlight text-txt-primary",
-        },
-        size: {
-            sm: "text-xs py-[2px] px-[6px]",
-            md: "text-sm py-[6px] px-2",
-        },
+  base: "font-medium rounded-lg w-max flex gap-1 items-center group whitespace-nowrap",
+  variants: {
+    variant: {
+      success: "bg-success text-txt-success",
+      warning: "bg-warning text-txt-warning",
+      danger: "bg-danger text-txt-danger",
+      info: "bg-info text-txt-info",
+      neutral: "bg-subtle text-txt-secondary",
+      outline: "text-txt-tertiary border border-border-low",
+      highlight: "bg-highlight text-txt-primary",
     },
-    defaultVariants: {
-        variant: "neutral",
-        size: "sm",
+    size: {
+      sm: "text-xs py-[2px] px-[6px]",
+      md: "text-sm py-[6px] px-2",
     },
+  },
+  defaultVariants: {
+    variant: "neutral",
+    size: "sm",
+  },
 });
 
 export function Badge(props: Props) {
-    const Component = props.asChild ? Slot : "div";
-    const { size, variant, ...restProps } = props;
-    return <Component {...restProps} className={badge({ size, variant })} />;
+  const Component = props.asChild ? Slot : "div";
+  const { className, size, variant, ...restProps } = props;
+  return (
+    <Component {...restProps} className={badge({ className, size, variant })} />
+  );
 }