small cleanup

bayesflow/networks/summary_network.py

@@ -28,7 +28,7 @@ def call(self, x: Tensor, **kwargs) -> Tensor:
     def compute_metrics(self, x: Tensor, stage: str = "training") -> dict[str, Tensor]:
         outputs = self(x, training=stage == "training")
 
-        metrics = {}
+        metrics = {"outputs": outputs}
 
         if self.base_distribution is not None:
             samples = self.base_distribution.sample((keras.ops.shape(x)[0],))
@@ -40,6 +40,4 @@ def compute_metrics(self, x: Tensor, stage: str = "training") -> dict[str, Tenso
                 for metric in self.metrics:
                     metrics[metric.name] = metric(outputs, samples)
 
-        metrics["outputs"] = outputs
-
         return metrics

bayesflow/simulators/hierarchical_simulator.py

@@ -6,8 +6,6 @@
 
 from .simulator import Simulator
 
-ShapeLike = Sequence[int | Sequence[int], ...]
-
 
 class HierarchicalSimulator(Simulator):
     def __init__(self, hierarchy: Sequence[Simulator]):