From 766a33afdf28f278f33187f99276642b4002aadb Mon Sep 17 00:00:00 2001
From: Bradlee Speice <bradlee@speice.io>
Date: Sat, 2 Mar 2024 13:42:41 -0500
Subject: [PATCH] Restrictions datatype

---
 spotify_model/track.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/spotify_model/track.py b/spotify_model/track.py
index 6a8dc69..d9c2743 100644
--- a/spotify_model/track.py
+++ b/spotify_model/track.py
@@ -16,7 +16,7 @@ class SimplifiedTrack(BaseModel):
     """
 
     artists: List[Dict[str, Union[Dict[str, str], str]]]
-    available_markets: List[str]
+    available_markets: Optional[List[str]]
     disc_number: int
     duration_ms: int
     explicit: bool
@@ -28,7 +28,7 @@ class SimplifiedTrack(BaseModel):
     linked_from: Optional[str]
     name: str
     preview_url: Optional[str]
-    restrictions: Optional[str]
+    restrictions: Optional[Union[str, Dict[str, str]]]
     track_number: int
     spotify_type: str = Field(alias="type")
     uri: str