From 3324a8caae84d25c1bd5e6948a00d70029f9e7b3 Mon Sep 17 00:00:00 2001
From: Marek Wolan <marek.wolan@methods.co.uk>
Date: Mon, 31 Jul 2023 18:54:29 +0000
Subject: [PATCH] Apply suggestions from code review

---
 benchmark/primaite_benchmark.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/benchmark/primaite_benchmark.py b/benchmark/primaite_benchmark.py
index 8a911720..9fec5711 100644
--- a/benchmark/primaite_benchmark.py
+++ b/benchmark/primaite_benchmark.py
@@ -411,7 +411,7 @@ def _plot_all_benchmarks_combined_session_av() -> Figure:
     return fig
 
 
-def run() -> NotImplementedError:
+def run() -> None:
     """Run the PrimAITE benchmark."""
     start_datetime = datetime.now()
     av_reward_per_episode_dicts = {}