Implement small code cleanup via a dict comprehension (#21)

bachya · Nov 21, 2022 · 534992e · 534992e
1 parent 2b8aa2d
commit 534992e
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/aiopurpleair/endpoints/__init__.py b/aiopurpleair/endpoints/__init__.py
@@ -49,15 +49,14 @@ async def _async_endpoint_request_with_models(
         Raises:
             InvalidRequestError: Raised on invalid parameters.
         """
-        payload: dict[str, Any] = {}
-
-        for api_query_param, func_param in query_param_map:
-            if not func_param:
-                continue
-            payload[api_query_param] = func_param
-
         try:
-            request = request_model.parse_obj(payload)
+            request = request_model.parse_obj(
+                {
+                    api_query_param: func_param
+                    for api_query_param, func_param in query_param_map
+                    if func_param is not None
+                }
+            )
         except ValidationError as err:
             raise InvalidRequestError(err) from err