set

PEtab-dev · Dec 11, 2024 · 4fd5e4c · 4fd5e4c
1 parent 04311e0
commit 4fd5e4c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/petab/v1/parameters.py b/petab/v1/parameters.py
@@ -3,7 +3,7 @@
 import numbers
 import warnings
 from collections import OrderedDict
-from collections.abc import Iterable, Sequence
+from collections.abc import Iterable, Sequence, Set
 from pathlib import Path
 from typing import (
     Literal,
@@ -258,7 +258,7 @@ def get_required_parameters_for_parameter_table(
     observable_df: pd.DataFrame,
     measurement_df: pd.DataFrame,
     mapping_df: pd.DataFrame = None,
-) -> set[str]:
+) -> Set[str]:
     """
     Get set of parameters which need to go into the parameter table
 

diff --git a/petab/v2/lint.py b/petab/v2/lint.py
@@ -4,6 +4,7 @@
 import logging
 from abc import ABC, abstractmethod
 from collections import OrderedDict
+from collections.abc import Set
 from dataclasses import dataclass, field
 from enum import IntEnum
 from pathlib import Path
@@ -688,7 +689,7 @@ def append_overrides(overrides):
 
 def get_required_parameters_for_parameter_table(
     problem: Problem,
-) -> set[str]:
+) -> Set[str]:
     """
     Get set of parameters which need to go into the parameter table