Skip to content

Commit 540fcf0

Browse files
committed
fix type-hint
1 parent 8bd9a34 commit 540fcf0

23 files changed

Lines changed: 69 additions & 46 deletions

src/gradient_free_optimizers/optimizer_search/bayesian_optimization.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Email: simon.blanke@yahoo.com
33
# License: MIT License
44

5-
from typing import List, Dict, Literal, Literal
5+
from typing import List, Dict, Literal, Union
66

77
from ..search import Search
88
from ..optimizers import BayesianOptimizer as _BayesianOptimizer
@@ -50,7 +50,8 @@ def __init__(
5050
self,
5151
search_space: Dict[str, list],
5252
initialize: Dict[
53-
Literal["grid", "vertices", "random", "warm_start"], int | List
53+
Literal["grid", "vertices", "random", "warm_start"],
54+
Union[int, list[dict]],
5455
] = {"grid": 4, "random": 2, "vertices": 4},
5556
constraints: List[callable] = [],
5657
random_state: int = None,

src/gradient_free_optimizers/optimizer_search/differential_evolution.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Email: simon.blanke@yahoo.com
33
# License: MIT License
44

5-
from typing import List, Dict, Literal, Literal
5+
from typing import List, Dict, Literal, Union
66

77
from ..search import Search
88
from ..optimizers import (
@@ -45,7 +45,8 @@ def __init__(
4545
self,
4646
search_space: Dict[str, list],
4747
initialize: Dict[
48-
Literal["grid", "vertices", "random", "warm_start"], int | List
48+
Literal["grid", "vertices", "random", "warm_start"],
49+
Union[int, list[dict]],
4950
] = {"grid": 4, "random": 2, "vertices": 4},
5051
constraints: List[callable] = [],
5152
random_state: int = None,

src/gradient_free_optimizers/optimizer_search/direct_algorithm.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Email: simon.blanke@yahoo.com
33
# License: MIT License
44

5-
from typing import List, Dict, Literal
5+
from typing import List, Dict, Literal, Union
66

77
from ..search import Search
88
from ..optimizers import DirectAlgorithm as _DirectAlgorithm
@@ -45,7 +45,8 @@ def __init__(
4545
self,
4646
search_space: Dict[str, list],
4747
initialize: Dict[
48-
Literal["grid", "vertices", "random", "warm_start"], int | List
48+
Literal["grid", "vertices", "random", "warm_start"],
49+
Union[int, list[dict]],
4950
] = {"grid": 4, "random": 2, "vertices": 4},
5051
constraints: List[callable] = [],
5152
random_state: int = None,

src/gradient_free_optimizers/optimizer_search/downhill_simplex.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Email: simon.blanke@yahoo.com
33
# License: MIT License
44

5-
from typing import List, Dict, Literal, Literal
5+
from typing import List, Dict, Literal, Union
66

77
from ..search import Search
88
from ..optimizers import DownhillSimplexOptimizer as _DownhillSimplexOptimizer
@@ -45,7 +45,8 @@ def __init__(
4545
self,
4646
search_space: Dict[str, list],
4747
initialize: Dict[
48-
Literal["grid", "vertices", "random", "warm_start"], int | List
48+
Literal["grid", "vertices", "random", "warm_start"],
49+
Union[int, list[dict]],
4950
] = {"grid": 4, "random": 2, "vertices": 4},
5051
constraints: List[callable] = [],
5152
random_state: int = None,

src/gradient_free_optimizers/optimizer_search/ensemble_optimizer.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Email: simon.blanke@yahoo.com
33
# License: MIT License
44

5-
from typing import List, Dict, Literal
5+
from typing import List, Dict, Literal, Union
66

77
from ..search import Search
88
from ..optimizers import EnsembleOptimizer as _EnsembleOptimizer
@@ -13,7 +13,8 @@ def __init__(
1313
self,
1414
search_space: Dict[str, list],
1515
initialize: Dict[
16-
Literal["grid", "vertices", "random", "warm_start"], int | List
16+
Literal["grid", "vertices", "random", "warm_start"],
17+
Union[int, list[dict]],
1718
] = {"grid": 4, "random": 2, "vertices": 4},
1819
constraints: List[callable] = [],
1920
random_state: int = None,

src/gradient_free_optimizers/optimizer_search/evolution_strategy.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Email: simon.blanke@yahoo.com
33
# License: MIT License
44

5-
from typing import List, Dict, Literal, Literal
5+
from typing import List, Dict, Literal, Union
66

77
from ..search import Search
88
from ..optimizers import (
@@ -51,7 +51,8 @@ def __init__(
5151
self,
5252
search_space: Dict[str, list],
5353
initialize: Dict[
54-
Literal["grid", "vertices", "random", "warm_start"], int | List
54+
Literal["grid", "vertices", "random", "warm_start"],
55+
Union[int, list[dict]],
5556
] = {"grid": 4, "random": 2, "vertices": 4},
5657
constraints: List[callable] = [],
5758
random_state: int = None,

src/gradient_free_optimizers/optimizer_search/forest_optimization.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Email: simon.blanke@yahoo.com
33
# License: MIT License
44

5-
from typing import List, Dict, Literal
5+
from typing import List, Dict, Literal, Union
66

77
from ..search import Search
88
from ..optimizers import ForestOptimizer as _ForestOptimizer
@@ -51,7 +51,8 @@ def __init__(
5151
self,
5252
search_space: Dict[str, list],
5353
initialize: Dict[
54-
Literal["grid", "vertices", "random", "warm_start"], int | List
54+
Literal["grid", "vertices", "random", "warm_start"],
55+
Union[int, list[dict]],
5556
] = {"grid": 4, "random": 2, "vertices": 4},
5657
constraints: List[callable] = [],
5758
random_state: int = None,

src/gradient_free_optimizers/optimizer_search/genetic_algorithm.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Email: simon.blanke@yahoo.com
33
# License: MIT License
44

5-
from typing import List, Dict, Literal
5+
from typing import List, Dict, Literal, Union
66

77
from ..search import Search
88
from ..optimizers import GeneticAlgorithmOptimizer as _GeneticAlgorithmOptimizer
@@ -49,7 +49,8 @@ def __init__(
4949
self,
5050
search_space: Dict[str, list],
5151
initialize: Dict[
52-
Literal["grid", "vertices", "random", "warm_start"], int | List
52+
Literal["grid", "vertices", "random", "warm_start"],
53+
Union[int, list[dict]],
5354
] = {"grid": 4, "random": 2, "vertices": 4},
5455
constraints: List[callable] = [],
5556
random_state: int = None,

src/gradient_free_optimizers/optimizer_search/grid_search.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Email: simon.blanke@yahoo.com
33
# License: MIT License
44

5-
from typing import List, Dict, Literal, Literal
5+
from typing import List, Dict, Literal, Union
66

77
from ..search import Search
88
from ..optimizers import GridSearchOptimizer as _GridSearchOptimizer
@@ -41,7 +41,8 @@ def __init__(
4141
self,
4242
search_space: Dict[str, list],
4343
initialize: Dict[
44-
Literal["grid", "vertices", "random", "warm_start"], int | List
44+
Literal["grid", "vertices", "random", "warm_start"],
45+
Union[int, list[dict]],
4546
] = {"grid": 4, "random": 2, "vertices": 4},
4647
constraints: List[callable] = [],
4748
random_state: int = None,

src/gradient_free_optimizers/optimizer_search/hill_climbing.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
# Email: simon.blanke@yahoo.com
33
# License: MIT License
44

5-
from typing import List, Dict, Literal, Literal
5+
from typing import List, Dict, Literal, Union
66

77
from ..search import Search
88
from ..optimizers import HillClimbingOptimizer as _HillClimbingOptimizer
@@ -44,7 +44,8 @@ def __init__(
4444
self,
4545
search_space: Dict[str, list],
4646
initialize: Dict[
47-
Literal["grid", "vertices", "random", "warm_start"], int | List
47+
Literal["grid", "vertices", "random", "warm_start"],
48+
Union[int, list[dict]],
4849
] = {"grid": 4, "random": 2, "vertices": 4},
4950
constraints: List[callable] = [],
5051
random_state: int = None,

0 commit comments

Comments
 (0)