|
1 | | -from .core_ext import ExprBuilder |
| 1 | +from .core_ext import ExprBuilder, VariableDomain |
2 | 2 | from .tupledict import make_tupledict |
3 | 3 |
|
4 | 4 | from collections.abc import Collection |
5 | | -from typing import Tuple, Union |
| 5 | +from typing import Tuple, Union, Optional |
6 | 6 |
|
7 | 7 |
|
8 | 8 | def make_variable_ndarray( |
9 | 9 | model, |
10 | 10 | shape: Union[Tuple[int, ...], int], |
11 | | - domain=None, |
12 | | - lb=None, |
13 | | - ub=None, |
14 | | - name=None, |
15 | | - start=None, |
| 11 | + domain: Optional[VariableDomain] = None, |
| 12 | + lb: Optional[float] = None, |
| 13 | + ub: Optional[float] = None, |
| 14 | + name: Optional[str] = None, |
| 15 | + start: Optional[float] = None, |
16 | 16 | ): |
17 | 17 | import numpy as np |
18 | 18 |
|
@@ -41,7 +41,13 @@ def make_variable_ndarray( |
41 | 41 |
|
42 | 42 |
|
43 | 43 | def make_variable_tupledict( |
44 | | - model, *coords: Collection, domain=None, lb=None, ub=None, name=None, start=None |
| 44 | + model, |
| 45 | + *coords: Collection, |
| 46 | + domain: Optional[VariableDomain] = None, |
| 47 | + lb: Optional[float] = None, |
| 48 | + ub: Optional[float] = None, |
| 49 | + name: Optional[str] = None, |
| 50 | + start: Optional[float] = None, |
45 | 51 | ): |
46 | 52 | kw_args = dict() |
47 | 53 | if domain is not None: |
|
0 commit comments