Remove unnecessary lines in schema.py

src/one-shot/schema.py

@@ -3,7 +3,6 @@ from datetime import datetime
 
 from pydantic import BaseModel
 
-
 class MultiWOZ(BaseModel):
     hotelArea: Optional[Literal[
         "center",
@@ -82,87 +81,4 @@ class MultiWOZ(BaseModel):
         "park",
         "swimmingpool",
         "theatre",
-    ]]
-
-
-class HeidelKBerg(BaseModel):
-    attractionArea: Optional[Literal[
-        "center",
-        "east",
-        "north",
-        "south",
-        "west",
-    ]]
-    attractionName: Optional[str]
-    attractionType: Optional[List[Literal[
-        "active",
-        "architecture", 
-        "child_friendly",
-        "educational",
-        "historic",
-        "indoor",
-        "nature",
-        "outdoor",
-        "passive",
-        "view",
-        "zoo", 
-    ]]]
-    attractionPricerange: Optional[Literal["cheap", "moderate", "expensive", "free"]]
-    hotelArea: Optional[Literal[
-        "center",
-        "east",
-        "north",
-        "south",
-        "west",
-    ]]
-    hotelBookday: Optional[Literal[
-        "monday", 
-        "tuesday",
-        "wednesday",
-        "thursday",
-        "friday",
-        "saturday", 
-        "sunday"
-    ]]
-    hotelBookpeople: Optional[int]
-    hotelBookstay: Optional[int]
-    hotelInternet: Optional[Literal["yes", "no"]]
-    hotelName: Optional[str]
-    hotelParking: Optional[Literal["limited", "yes", "no"]]
-    hotelPricerange: Optional[Literal["cheap", "moderate", "expensive"]]
-    hotelStars: Optional[Literal["0", "1", "2", "3", "4", "5"]]
-    restaurantArea: Optional[Literal[
-        "center",
-        "east",
-        "north",
-        "south",
-        "west",
-    ]]
-    restaurantBookday: Optional[Literal[
-        "monday", 
-        "tuesday",
-        "wednesday",
-        "thursday",
-        "friday",
-        "saturday", 
-        "sunday"
-    ]]
-    restaurantBookpeople: Optional[int]
-    restaurantFood: Optional[List[Literal[
-        "african",
-        "american",
-        "asian",
-        "chinese",
-        "german",
-        "greek",
-        "indian",
-        "international",
-        "italian",
-        "japanese",
-        "korean",
-        "oriental",
-        "sushi",
-        "thai"
-    ]]]
-    restaurantName: Optional[str]
-    restaurantPricerange: Optional[Literal["cheap", "moderate", "expensive"]]
\ No newline at end of file
+    ]]
\ No newline at end of file