Add arviz version to InferenceData attrs (#1086)

* Add arviz version to InferenceData attrs * arviz version to attr
arviz-devs · Feb 23, 2020 · dc20591 · dc20591
1 parent 148b549
commit dc20591
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -19,6 +19,7 @@
 * **Experimental Feature**: Added `arviz.wrappers` module to allow ArviZ to
   refit the models if necessary
 * **Experimental Feature**: Added `reloo` function to ArviZ
+* ArviZ version to InferenceData attributes. (#1086)
 
 
 ### Maintenance and fixes

diff --git a/arviz/data/base.py b/arviz/data/base.py
@@ -7,6 +7,7 @@
 import xarray as xr
 
 from .. import utils
+from .. import __version__
 
 CoordSpec = Dict[str, List[Any]]
 DimSpec = Dict[str, List[str]]
@@ -215,7 +216,10 @@ def make_attrs(attrs=None, library=None):
     dict
         attrs
     """
-    default_attrs = {"created_at": datetime.datetime.utcnow().isoformat()}
+    default_attrs = {
+        "created_at": datetime.datetime.utcnow().isoformat(),
+        "arviz_version": __version__,
+    }
     if library is not None:
         library_name = library.__name__
         default_attrs["inference_library"] = library_name