From 87284007604f07b7ccb8cd7ac0c85bfe38fe143a Mon Sep 17 00:00:00 2001
From: Cola Cheng <imlikedog@gmail.com>
Date: Sat, 10 Dec 2022 15:06:09 +0800
Subject: [PATCH] Align WebSocket.Frame.encode/2 spec with WebSocket.encode/2.

---
 lib/mint/web_socket/frame.ex | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/mint/web_socket/frame.ex b/lib/mint/web_socket/frame.ex
index fc96150e..7ba1df2c 100644
--- a/lib/mint/web_socket/frame.ex
+++ b/lib/mint/web_socket/frame.ex
@@ -93,7 +93,7 @@ defmodule Mint.WebSocket.Frame do
 
   def new_mask, do: :crypto.strong_rand_bytes(4)
 
-  @spec encode(Mint.WebSocket.t(), Mint.WebSocket.frame()) ::
+  @spec encode(Mint.WebSocket.t(), Mint.WebSocket.shorthand_frame() | Mint.WebSocket.frame()) ::
           {:ok, Mint.WebSocket.t(), bitstring()}
           | {:error, Mint.WebSocket.t(), WebSocketError.t()}
   def encode(websocket, frame) when is_friendly_frame(frame) do