Skip to content

Commit 4e09b8b

Browse files
committed
fix "ContinuousApproximator has no call method"
1 parent 3435038 commit 4e09b8b

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

bayesflow/networks/time_series_network/skip_recurrent.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,17 +41,20 @@ def __init__(
4141

4242
recurrent_constructor = find_recurrent_net(recurrent_type)
4343

44-
self.recurrent = recurrent_constructor(
44+
recurrent = recurrent_constructor(
4545
units=hidden_dim // 2 if bidirectional else hidden_dim,
4646
dropout=dropout,
4747
)
48-
self.skip_recurrent = recurrent_constructor(
48+
skip_recurrent = recurrent_constructor(
4949
units=hidden_dim // 2 if bidirectional else hidden_dim,
5050
dropout=dropout,
5151
)
5252
if bidirectional:
53-
self.recurrent = keras.layers.Bidirectional(self.recurrent)
54-
self.skip_recurrent = keras.layers.Bidirectional(self.skip_recurrent)
53+
recurrent = keras.layers.Bidirectional(recurrent)
54+
skip_recurrent = keras.layers.Bidirectional(skip_recurrent)
55+
56+
self.recurrent = recurrent
57+
self.skip_recurrent = skip_recurrent
5558
self.input_channels = input_channels
5659

5760
def call(self, time_series: Tensor, training: bool = False, **kwargs) -> Tensor:

0 commit comments

Comments
 (0)