def get_schema(dataset):
    """Get schema for a dataset."""
    if dataset == "multiwoz":
        schema = {
            "attraction": {
                "book": [],
                "semi": ["area", "name", "type"]
            },
            "hotel": {
                "book": ["day", "people", "stay"],
                "semi": ["area", "internet", "name", "parking", "pricerange", "stars", "type"]
            },
            "restaurant": {
                "book": ["day", "people", "time"],
                "semi": ["area", "food", "name", "pricerange"]
            },
            "taxi": {
                "book": [],
                "semi": ["arriveby", "departure", "destination", "leaveat"]
            },
            "train": {
                "book": ["people"],
                "semi": ["arriveby", "day", "departure", "destination", "leaveat"]
            }
        }
    elif dataset == "woz":
        schema = {
            "restaurant": {
                "book": [],
                "semi": ["area", "food", "pricerange"]
            }
        }
    else:
        raise ValueError(f"Unknown dataset: `{dataset}`")
    return schema