1 files changed, 2 insertions, 2 deletions
diff --git a/src/eval.py b/src/eval.py
index 9d55437..4485523 100644
--- a/src/eval.py
+++ b/src/eval.py
@@ -91,8 +91,8 @@ def halting_report(model, cfg: Config) -> dict:
     gaps, steps = [], []
     for n in range(cfg.range_start, cfg.range_end + 1):
         x = torch.tensor(encode_int(n, cfg), dtype=torch.long).unsqueeze(0)
-        h = model._initial_state(x)
-        _, s = model._run_cell(h)
+        h = model._encode(x)
+        _, s = model._run_compute(h)
         gaps.append(next_prime(n, primes) - n)
         steps.append(float(s.mean()))
     mean = float(np.mean(steps))