fix whitespace in docs, always return list from _batched_response

ignore doctest outputs

Author: snbianco

astroquery/mast/missions.py

@@ -444,10 +444,12 @@ def get_product_list_async(self, datasets):
             max_batch=1000,
             param_key="dataset_ids",
             request_func=lambda p: self._service_api_connection.missions_request_async(self.service, p),
-            extract_func=lambda r: r,  # missions_request_async already returns one result
+            extract_func=lambda r: [r],  # missions_request_async already returns one result
             desc=f"Fetching products for {len(datasets)} unique datasets"
         )
-        return results
+
+        # Return a list of responses only if multiple requests were made
+        return results[0] if len(results) == 1 else results
 
     def get_unique_product_list(self, datasets):
         """

astroquery/mast/utils.py

@@ -135,7 +135,6 @@ def _batched_request(
 
                 # Extend results with new response
                 new_resp = extract_func(resp)
-                new_resp = new_resp if isinstance(new_resp, list) else [new_resp]
                 results.extend(new_resp)
 
                 # Update progress bar

docs/mast/mast.rst

@@ -116,31 +116,31 @@ across multiple requests, with results combined into the final return object.
    ...
    >>> # Resolve a single object
    >>> coords = mast.resolve_object("M101", resolver="NED")
-   >>> print(coords)
+   >>> print(coords)  # doctest: +IGNORE_OUTPUT
    <SkyCoord (ICRS): (ra, dec) in deg
-    (210.80227, 54.34895)>
+       (210.80227, 54.34895)>
    ...
    >>> # Resolve multiple objects
    >>> coords_multi = mast.resolve_object(["M101", "M51"], resolver="SIMBAD")
-   >>> pprint(coords_multi)
+   >>> pprint(coords_multi)  # doctest: +IGNORE_OUTPUT
    {'M101': <SkyCoord (ICRS): (ra, dec) in deg
        (210.802429, 54.34875)>,
     'M51': <SkyCoord (ICRS): (ra, dec) in deg
        (202.469575, 47.195258)>}
    ...
    >>> # Resolve a single object with all resolvers
    >>> coords_dict = mast.resolve_object("M101", resolve_all=True)
-   >>> pprint(coords_dict)
+   >>> pprint(coords_dict)  # doctest: +IGNORE_OUTPUT
    {'NED': <SkyCoord (ICRS): (ra, dec) in deg
-      (210.80227, 54.34895)>,
-   'SIMBAD': <SkyCoord (ICRS): (ra, dec) in deg
-      (210.802429, 54.34875)>,
-   'SIMBADCFA': <SkyCoord (ICRS): (ra, dec) in deg
-      (210.802429, 54.34875)>}
+       (210.80227, 54.34895)>,
+    'SIMBAD': <SkyCoord (ICRS): (ra, dec) in deg
+       (210.802429, 54.34875)>,
+    'SIMBADCFA': <SkyCoord (ICRS): (ra, dec) in deg
+       (210.802429, 54.34875)>}
    ...
    >>> # Resolve multiple objects with all resolvers
    >>> coords_dict_multi = mast.resolve_object(["M101", "M51"], resolve_all=True)
-   >>> pprint(coords_dict_multi)
+   >>> pprint(coords_dict_multi)  # doctest: +IGNORE_OUTPUT
    {'M101': {'NED': <SkyCoord (ICRS): (ra, dec) in deg
       (210.80227, 54.34895)>,
             'SIMBAD': <SkyCoord (ICRS): (ra, dec) in deg