Skip to content

Commit 32b5a03

Browse files
committed
add failing tests for TimeSeriesTransformer
The deserialization of the `TimeSeriesTransformer` is broken, but it went undetected as this network is not included in the tests yet. This commit adds the network, but does not resolve the deserialization problems yet. Regards #423
1 parent 65dee2c commit 32b5a03

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

tests/test_networks/conftest.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,13 @@ def time_series_network(summary_dim):
119119
return TimeSeriesNetwork(summary_dim=summary_dim)
120120

121121

122+
@pytest.fixture(scope="function")
123+
def time_series_transformer(summary_dim):
124+
from bayesflow.networks import TimeSeriesTransformer
125+
126+
return TimeSeriesTransformer(summary_dim=summary_dim)
127+
128+
122129
@pytest.fixture(scope="function")
123130
def set_transformer(summary_dim):
124131
from bayesflow.networks import SetTransformer
@@ -133,7 +140,9 @@ def deep_set(summary_dim):
133140
return DeepSet(summary_dim=summary_dim)
134141

135142

136-
@pytest.fixture(params=[None, "time_series_network", "set_transformer", "deep_set"], scope="function")
143+
@pytest.fixture(
144+
params=[None, "time_series_network", "time_series_transformer", "set_transformer", "deep_set"], scope="function"
145+
)
137146
def summary_network(request, summary_dim):
138147
if request.param is None:
139148
return None

0 commit comments

Comments
 (0)