File tree Expand file tree Collapse file tree 4 files changed +6
-6
lines changed Expand file tree Collapse file tree 4 files changed +6
-6
lines changed Original file line number Diff line number Diff line change 11
11
# express or implied. See the License for the specific language governing
12
12
# permissions and limitations under the License.
13
13
14
- from typing import Tuple
14
+ from typing import Optional , Tuple
15
15
16
16
import torch
17
17
from torch import nn
@@ -87,7 +87,7 @@ def __init__(
87
87
hidden_dimension : int ,
88
88
distr_output = StudentTOutput (),
89
89
kernel_size : int = 25 ,
90
- scaling : str = "mean" ,
90
+ scaling : Optional [ str ] = "mean" ,
91
91
) -> None :
92
92
super ().__init__ ()
93
93
Original file line number Diff line number Diff line change @@ -55,7 +55,7 @@ def __init__(
55
55
activation : str ,
56
56
norm_first : bool ,
57
57
num_encoder_layers : int ,
58
- scaling : str ,
58
+ scaling : Optional [ str ] ,
59
59
lags_seq : Optional [List [int ]] = None ,
60
60
distr_output = StudentTOutput (),
61
61
) -> None :
Original file line number Diff line number Diff line change @@ -108,7 +108,7 @@ def __init__(
108
108
activation : str ,
109
109
norm_first : bool ,
110
110
num_encoder_layers : int ,
111
- scaling : str ,
111
+ scaling : Optional [ str ] ,
112
112
distr_output = StudentTOutput (),
113
113
) -> None :
114
114
super ().__init__ ()
Original file line number Diff line number Diff line change 11
11
# express or implied. See the License for the specific language governing
12
12
# permissions and limitations under the License.
13
13
14
- from typing import List , Tuple
14
+ from typing import List , Optional , Tuple
15
15
16
16
import torch
17
17
from torch import nn
@@ -243,7 +243,7 @@ def __init__(
243
243
num_layers_decoder : int ,
244
244
layer_norm : bool ,
245
245
distr_output : Output ,
246
- scaling : str ,
246
+ scaling : Optional [ str ] ,
247
247
) -> None :
248
248
super ().__init__ ()
249
249
You can’t perform that action at this time.
0 commit comments