refactor: clean up mypy errors

Author: dustinbyrne

posthog/client.py

@@ -227,8 +227,8 @@ def __init__(
         self.timeout = timeout
         self._feature_flags = None  # private variable to store flags
         self.feature_flags_by_key = None
-        self.group_type_mapping = None
-        self.cohorts = None
+        self.group_type_mapping: Optional[dict[str, str]] = None
+        self.cohorts: Optional[dict[str, Any]] = None
         self.poll_interval = poll_interval
         self.feature_flags_request_timeout_seconds = (
             feature_flags_request_timeout_seconds
@@ -1213,9 +1213,9 @@ def _update_flag_state(
         self, data: FlagDefinitionCacheData, old_flags_by_key: Optional[dict] = None
     ) -> None:
         """Update internal flag state from cache data and invalidate evaluation cache if changed."""
-        self.feature_flags = data.get("flags") or []
-        self.group_type_mapping = data.get("group_type_mapping") or {}
-        self.cohorts = data.get("cohorts") or {}
+        self.feature_flags = data.get("flags", [])
+        self.group_type_mapping = data.get("group_type_mapping", {})
+        self.cohorts = data.get("cohorts", {})
 
         # Invalidate evaluation cache if flag definitions changed
         if (
@@ -1417,7 +1417,8 @@ def _compute_flag_locally(
         flag_filters = feature_flag.get("filters") or {}
         aggregation_group_type_index = flag_filters.get("aggregation_group_type_index")
         if aggregation_group_type_index is not None:
-            group_name = self.group_type_mapping.get(str(aggregation_group_type_index))
+            group_type_mapping = self.group_type_mapping or {}
+            group_name = group_type_mapping.get(str(aggregation_group_type_index))
 
             if not group_name:
                 self.log.warning(