Skip to content

Commit

Permalink
fix dict comprehension bug in bf.networks.transformers.set_transforme…
Browse files Browse the repository at this point in the history
…r.py
  • Loading branch information
marvinschmitt authored Aug 13, 2024
1 parent b4afea0 commit 5c61054
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion bayesflow/networks/transformers/set_transformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -70,7 +70,7 @@ def __init__(
if num_inducing_points is not None:
block = InducedSetAttentionBlock(**attention_block_settings)
else:
block = SetAttentionBlock(**{k: v for k, v in attention_block_settings if k != "num_inducing_points"})
block = SetAttentionBlock(**{k: v for k, v in attention_block_settings.items() if k != "num_inducing_points"})
self.attention_blocks.add(block)

# Pooling will be applied as a final step to the abstract representations obtained from set attention
Expand Down

0 comments on commit 5c61054

Please sign in to comment.