Fix hybrid bug

thebuckleylab · Jun 18, 2024 · c7f43f5 · c7f43f5
1 parent 1f22fa5
commit c7f43f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jpc/_test.py b/jpc/_test.py
@@ -128,7 +128,7 @@ def test_hpc(
           atol=1e-3
       ),
       dt: Union[float, int] = None
-):
+) -> Tuple[Scalar, Scalar, Scalar, Array]:
     """Computes test metrics for hybrid predictive coding.
 
     Calculates input accuracy of (i) amortiser, (ii) generative, and (ii)
@@ -171,7 +171,7 @@ def test_hpc(
     amort_preds = amort_activities[0]
     hpc_preds = solve_pc_activities(
         network=generator,
-        activities=amort_activities[1:],
+        activities=amort_activities,
         output=output,
         solver=solver,
         n_iters=n_iters,