Skip to content

Commit b87f42d

Browse files
committed
fix
1 parent 9653e9d commit b87f42d

File tree

1 file changed

+13
-21
lines changed

1 file changed

+13
-21
lines changed

cognite/client/data_classes/simulators/filters.py

Lines changed: 13 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,14 @@
88
from cognite.client.utils.useful_types import SequenceNotStr
99

1010

11+
def parse_str_or_sequence(value: str | Sequence[str] | None) -> list[str] | None:
12+
if isinstance(value, str):
13+
return [value]
14+
elif isinstance(value, Sequence):
15+
return list(value)
16+
return None
17+
18+
1119
class SimulatorIntegrationFilter(CogniteFilter):
1220
def __init__(
1321
self,
@@ -52,27 +60,11 @@ def __init__(
5260
model_revision_external_ids: str | Sequence[str] | None = None,
5361
) -> None:
5462
self.model_external_ids = [model_external_ids] if isinstance(model_external_ids, str) else model_external_ids
55-
self.simulator_integration_external_ids = (
56-
[simulator_integration_external_ids]
57-
if isinstance(simulator_integration_external_ids, str)
58-
else simulator_integration_external_ids
59-
)
60-
self.simulator_external_ids = (
61-
[simulator_external_ids] if isinstance(simulator_external_ids, str) else simulator_external_ids
62-
)
63-
self.routine_external_ids = (
64-
[routine_external_ids] if isinstance(routine_external_ids, str) else routine_external_ids
65-
)
66-
self.routine_revision_external_ids = (
67-
[routine_revision_external_ids]
68-
if isinstance(routine_revision_external_ids, str)
69-
else routine_revision_external_ids
70-
)
71-
self.model_revision_external_ids = (
72-
[model_revision_external_ids]
73-
if isinstance(model_revision_external_ids, str)
74-
else model_revision_external_ids
75-
)
63+
self.simulator_integration_external_ids = parse_str_or_sequence(simulator_integration_external_ids)
64+
self.simulator_external_ids = parse_str_or_sequence(simulator_external_ids)
65+
self.routine_external_ids = parse_str_or_sequence(routine_external_ids)
66+
self.routine_revision_external_ids = parse_str_or_sequence(routine_revision_external_ids)
67+
self.model_revision_external_ids = parse_str_or_sequence(model_revision_external_ids)
7668
self.status = status
7769
self.run_type = run_type
7870

0 commit comments

Comments
 (0)