Skip to content

Commit

Permalink
typing
Browse files Browse the repository at this point in the history
vinicvaz committed Nov 22, 2023
1 parent 79f4393 commit 1fa898e
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion pieces/ToyDatasetsPiece/models.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from pydantic import BaseModel, Field
from enum import Enum
from typing import List

class DatasetName(str, Enum):
iris = "iris"
@@ -13,4 +14,4 @@ class InputModel(BaseModel):
dataset: DatasetName = Field(default='iris', title='Dataset name')

class OutputModel(BaseModel):
data: list[dict]
data: List[dict]

0 comments on commit 1fa898e

Please sign in to comment.