Skip to content

Commit 3cee5ae

Browse files
committed
Update SequenceNetwork to SequentialNetwork
1 parent 17551f5 commit 3cee5ae

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

bayesflow/networks.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
DeepSet,
2525
HierarchicalNetwork,
2626
InvariantNetwork,
27+
SequenceNetwork,
2728
SequentialNetwork,
2829
SetTransformer,
2930
SplitNetwork,

tests/test_summary_networks.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525
from bayesflow.summary_networks import (
2626
DeepSet,
2727
HierarchicalNetwork,
28-
SequentialNetwork,
28+
SequenceNetwork,
2929
SetTransformer,
3030
TimeSeriesTransformer,
3131
)
@@ -110,8 +110,8 @@ def test_deep_set(num_equiv, summary_dim):
110110

111111
@pytest.mark.parametrize("num_conv_layers", [1, 3])
112112
@pytest.mark.parametrize("lstm_units", [16, 32])
113-
def test_sequential_network(num_conv_layers, lstm_units):
114-
"""Tests the fidelity of the ``SequentialNetwork`` w.r.t. shape integrity
113+
def test_sequence_network(num_conv_layers, lstm_units):
114+
"""Tests the fidelity of the ``SequenceNetwork`` w.r.t. shape integrity
115115
using a number of relevant configurations."""
116116

117117
# Create settings dict and network
@@ -120,7 +120,7 @@ def test_sequential_network(num_conv_layers, lstm_units):
120120
"num_conv_layers": num_conv_layers,
121121
"lstm_units": lstm_units,
122122
}
123-
net = SequentialNetwork(**settings)
123+
net = SequenceNetwork(**settings)
124124

125125
# Create test data and pass through network
126126
x, _, _ = _gen_randomized_3d_data()

0 commit comments

Comments
 (0)