self.annotation_select_mask might be None

Author: Photonios

psqlextra/query.py

@@ -48,7 +48,7 @@ def rename_annotations(self, annotations) -> None:
 
             if django.VERSION < (2, 0):
                 self.set_annotation_mask(
-                    (new_name if v == old_name else v for v in self.annotation_select_mask))
+                    (new_name if v == old_name else v for v in (self.annotation_select_mask or [])))
 
     def add_join_conditions(self, conditions: Dict[str, Any]) -> None:
         """Adds an extra condition to an existing JOIN.