Keras ValueError re-appearing in newest package version?

[floralynth]

Hi all, I seem to have bumped into an issue previously seen as resolved:

Keras ValueError: For a `build()` method with more than one argument, all arguments should have a `_shape` suffix and match an argument from `call()`

Hello, I’m encountering this error when attempting to run the minimal working example on current github readme. For troubleshooting, I’ve attempted bayesflow 2.0.7 and 2.0.6, jax GPU, jax CPU, tensorflow CPU backends, and downgrading keras to 3.9.0, with no luck. Would be happy to try any suggestions you may have. Install/setup details: Running from linux (RHEL 9.2 on university’s hpc) uv for install/package management Install: uv venv test-bf source test-bf/bin/activate uv pip install b…

As discussed in that post, using SetTransformer or DeepSet results in an error once workflow.fit() is called:

ValueError: For a `build()` method with more than one argument, all arguments should have a `_shape` suffix and match an argument from `call()`. E.g. `build(self, foo_shape, bar_shape)`  For layer 'SetAttention', Received `build()` argument `self`, which does not end in `_shape`.

I am pretty new to code discussion forums etc, but my attempt at a minimal reproducible example (it errors before initializing, not sure it would run properly if it did but gets the point across?):

    
workflow = bf.BasicWorkflow(
    inference_network=bf.networks.FlowMatching(),
    inference_variables=["parameters"],
    summary_variables=["observables"],
    summary_network=bf.networks.SetTransformer(),
    simulator=bf.simulators.SIR(),
)

history = workflow.fit_online(
    epochs=2,
    batch_size=32,
    num_batches_per_epoch=200,
)

Bayesflow version: 2.0.11

Keras 3.14.0

Python 3.11

Happy to provide more info as guided by the community! Thanks for all the help