Skip to content

Commit

Permalink
update type
Browse files Browse the repository at this point in the history
  • Loading branch information
vinicvaz committed May 7, 2024
1 parent 658f2f8 commit 5430850
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions pieces/ProphetTrainModelPiece/models.py
Original file line number Diff line number Diff line change
@@ -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


Expand All @@ -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."
)
Expand Down

0 comments on commit 5430850

Please sign in to comment.