diff options
Diffstat (limited to 'generalresearch/models/precision/task_collection.py')
| -rw-r--r-- | generalresearch/models/precision/task_collection.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/generalresearch/models/precision/task_collection.py b/generalresearch/models/precision/task_collection.py index e2942b5..bd7b274 100644 --- a/generalresearch/models/precision/task_collection.py +++ b/generalresearch/models/precision/task_collection.py @@ -1,7 +1,7 @@ -from typing import List +from typing import Any, Dict, List import pandas as pd -from pandera import Column, DataFrameSchema, Check, Index +from pandera import Check, Column, DataFrameSchema, Index from generalresearch.locales import Localelator from generalresearch.models.precision import PrecisionStatus @@ -56,7 +56,7 @@ class PrecisionTaskCollection(TaskCollection): items: List[PrecisionSurvey] _schema = PrecisionTaskCollectionSchema - def to_row(self, s: PrecisionSurvey): + def to_row(self, s: PrecisionSurvey) -> Dict[str, Any]: d = s.model_dump( mode="json", exclude={ |