fix flow matching fixture

LarsKue · LarsKue · commit fbd9a8ba1e0b · 2025-04-08T12:13:16.000-04:00
diff --git a/tests/test_networks/conftest.py b/tests/test_networks/conftest.py
@@ -22,7 +22,7 @@ def flow_matching():
     from bayesflow.networks import FlowMatching
 
     return FlowMatching(
-        subnet_kwargs={"widths": None, "width": 64, "depth": 2},
+        subnet_kwargs={"widths": [64, 64]},
         integrate_kwargs={"method": "rk45", "steps": 100},
     )
 

Original file line number	Diff line number	Diff line change
`@@ -22,7 +22,7 @@ def flow_matching():`
`22`	`22`	`from bayesflow.networks import FlowMatching`
`23`	`23`
`24`	`24`	`return FlowMatching(`
`25`		`- subnet_kwargs={"widths": None, "width": 64, "depth": 2},`
	`25`	`+ subnet_kwargs={"widths": [64, 64]},`
`26`	`26`	`integrate_kwargs={"method": "rk45", "steps": 100},`
`27`	`27`	`)`
`28`	`28`