Skip to content

Commit c4005b6

Browse files
committed
Added missing mypy_fix decorators
1 parent a404a1b commit c4005b6

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

src/omnipy/modules/general/tasks.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ def import_directory(directory: str,
5656
_DatasetT = TypeVar('_DatasetT', bound=Dataset)
5757

5858

59+
@mypy_fix_task_template
5960
@TaskTemplate
6061
def convert_dataset(dataset: Dataset, dataset_cls: type[_DatasetT]) -> _DatasetT:
6162
return dataset_cls(dataset)

src/omnipy/modules/isa/flows.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,13 @@
11
from omnipy.compute.flow import LinearFlowTemplate
2+
from omnipy.compute.typing import mypy_fix_linear_flow_template
23

34
from ..general.tasks import convert_dataset
45
from ..json.datasets import JsonDictOfDictsDataset
56
from ..json.flows import flatten_nested_json, transpose_dict_of_dicts_2_list_of_dicts
67
from .datasets import FlattenedIsaJsonDataset, IsaJsonDataset
78

89

10+
@mypy_fix_linear_flow_template
911
@LinearFlowTemplate(
1012
convert_dataset.refine(fixed_params=dict(dataset_cls=JsonDictOfDictsDataset)),
1113
transpose_dict_of_dicts_2_list_of_dicts,

0 commit comments

Comments
 (0)