Citations allow passing single function (#89)

* allow passing single function, update refs

* fix doctring

Author: aloctavodia

src/arviz_base/citations.py

@@ -10,7 +10,7 @@ def citations(methods=None, filepath=None, format_type="bibtex"):
 
     Parameters
     ----------
-    methods : List
+    methods : Callable or list of callable, optional
         Methods implemented in ArviZ from which to retrieve citations.
     filepath : str, optional
         Specifies the location to save the file with the citations.
@@ -28,6 +28,10 @@ def citations(methods=None, filepath=None, format_type="bibtex"):
     """
     method_citations = [{"doi": "10.21105/joss.XXXXX"}]
     if methods is not None:
+        if isinstance(methods, str):
+            raise TypeError("you should pass an ArviZ function or list of functions.")
+        if not isinstance(methods, list | tuple):
+            methods = [methods]
         for method in methods:
             _extract_ids_per_entry(method_citations, method.__doc__)
 

src/arviz_base/citations.pyi

@@ -2,12 +2,14 @@
 
 import os
 import re
-from typing import List
+from collections.abc import Callable
 
 from _typeshed import Incomplete
 
 def citations(
-    methods: List | None = ..., filepath: str | None = ..., format_type: str = ...
+    methods: Callable | list[Callable] | None = ...,
+    filepath: str | None = ...,
+    format_type: str = ...,
 ) -> None: ...
 def _extract_ids_per_entry(data: Incomplete, text: Incomplete) -> None: ...
 def _find_bibtex_entries(header: Incomplete, data: Incomplete) -> None: ...

src/arviz_base/references.bib

@@ -290,3 +290,15 @@ @misc{bickel2005
 	doi= {https://doi.org/10.48550/arXiv.math/0505419},
 	url={https://arxiv.org/abs/math/0505419},
 }
+
+
+@misc{sivula_2025,
+	title={Uncertainty in Bayesian Leave-One-Out Cross-Validation Based Model Comparison},
+	author={Tuomas Sivula and Måns Magnusson and Asael Alonzo Matamoros and Aki Vehtari},
+	year={2025},
+	eprint={2008.10296},
+	archivePrefix={arXiv},
+	primaryClass={stat.ME},
+	doi={https://doi.org/10.48550/arXiv.2008.10296},
+	url={https://arxiv.org/abs/2008.10296},
+}