Fix prefetching for include param fields

gundotio · Jan 13, 2023 · ec2edbd · ec2edbd
1 parent 451a46c
commit ec2edbd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/worf/views/list.py b/worf/views/list.py
@@ -143,7 +143,8 @@ def get_processed_queryset(self):
                 )
 
         if self.include_fields and self.bundle.get("include"):
-            for item in set(self.include_fields.keys()) & set(self.bundle["include"]):
+            include = field_list(self.bundle["include"])
+            for item in set(self.include_fields.keys()) & set(include):
                 if isinstance(self.include_fields[item], Prefetch):
                     queryset = queryset.prefetch_related(self.include_fields[item])
                 elif isinstance(self.include_fields[item], str):