From 5430850411f51a5c83fba9b3cf1b3da276af8245 Mon Sep 17 00:00:00 2001 From: vinicvaz Date: Tue, 7 May 2024 15:18:04 -0300 Subject: [PATCH] update type --- pieces/ProphetTrainModelPiece/models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pieces/ProphetTrainModelPiece/models.py b/pieces/ProphetTrainModelPiece/models.py index c642cf5..e84a0f2 100644 --- a/pieces/ProphetTrainModelPiece/models.py +++ b/pieces/ProphetTrainModelPiece/models.py @@ -1,6 +1,6 @@ from pydantic import BaseModel, Field from enum import Enum -from typing import List +from typing import List, Optional from datetime import date @@ -24,7 +24,7 @@ class InputModel(BaseModel): default=GrowthTrend.linear, description="The growth trend of the data. Options are `linear`, `logistic` and `flat`. Default is `linear`." ) - changepoints: List[date] = Field( + changepoints: Optional[List[date]] = Field( default=None, description="List of dates at which to include potential changepoints. If not specified, potential changepoints are selected automatically." )