From 840274d935a25c5d769578b3fe3921e80797da71 Mon Sep 17 00:00:00 2001
From: klashenriksson <6956114+klashenriksson@users.noreply.github.com>
Date: Thu, 7 Dec 2023 09:51:03 +0100
Subject: [PATCH] report inf step if requested

---
 emote/memory/memory.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/emote/memory/memory.py b/emote/memory/memory.py
index 6f2fea7e..f48312d5 100644
--- a/emote/memory/memory.py
+++ b/emote/memory/memory.py
@@ -312,6 +312,8 @@ def get_report(
                 out_lists[key] = list(window)
                 out[key] = sum(window) / len(window)
                 out[f"{key}/cumulative"] = self.windowed_scalar_cumulative[key]
+            elif key == "inf_step":
+                out[key] = self.completed_inferences
 
         return out, out_lists