Slight aesthetics

bayesflow-org · Jun 4, 2024 · e65cd94 · e65cd94
1 parent fadd718
commit e65cd94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bayesflow/experimental/networks/deep_set/invariant_module.py b/bayesflow/experimental/networks/deep_set/invariant_module.py
@@ -94,7 +94,7 @@ def call(self, input_set: Tensor, **kwargs) -> Tensor:
 
         set_summary = self.inner_fc(input_set, training=kwargs.get("training", False))
         set_summary = self.pooling_layer(set_summary, training=kwargs.get("training", False))
-        set_summary= self.outer_fc(set_summary, training=kwargs.get("training", False))
+        set_summary = self.outer_fc(set_summary, training=kwargs.get("training", False))
         return set_summary
 
     def build(self, input_shape):