Add typing to make_csv dictionary

DiamondLightSource · Aug 13, 2024 · 3fdb4c6 · 3fdb4c6
1 parent d22622d
commit 3fdb4c6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/mx_bluesky/bimorph_optimisation_plan/data_saver.py b/src/mx_bluesky/bimorph_optimisation_plan/data_saver.py
@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Callable
+from typing import Callable, Dict
 
 
 def generate_filename(
@@ -41,7 +41,7 @@ def make_csv(docs: list) -> str:
  Returms:
  A string of given list's csv equivalent
  """
- csv_dict = {}
+ csv_dict: Dict[str, list] = {}
  headers = []
 
  for doc in docs: