last additional typing

bptlab · May 24, 2024 · 56b2e0c · 56b2e0c
1 parent 35b0441
commit 56b2e0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tracex_project/extraction/logic/orchestrator.py b/tracex_project/extraction/logic/orchestrator.py
@@ -128,7 +128,7 @@ def get_data(self):
         """Return the data for the orchestrator instance."""
         return self.data
 
-    def set_cohort(self, cohort):
+    def set_cohort(self, cohort: Dict[str, str] | None):
         """Set the cohort for the orchestrator instance."""
         self.cohort = cohort