Fix attrs (#242)

* fixing deletion of attributes in dataset varibles * Change documented in history.md
openradar · Nov 16, 2024 · b4684fa · b4684fa
1 parent dc76619
commit b4684fa
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/docs/history.md b/docs/history.md
@@ -4,6 +4,7 @@
 
 * ENH: Adding test to `open_datatree` function for all backends. Adding "scan_name" to nexradlevel2 datatree attributes ({pull}`238`) by [@aladinor](https://github.com/aladinor)
 * FIX: Improving performance of `open_nexradlevel2_datatree` function and adding tests for `sweep` parameter. ({issue}`239`) ({pull}`240`) by [@aladinor](https://github.com/aladinor)
+* FIX: Keeping attributes at each variable when using `open_nexradlevel2_datatree`. ({issue}`241`) ({pull}`242`) by [@aladinor](https://github.com/aladinor)
 
 ## 0.8.0 (2024-11-04)
 

diff --git a/xradar/io/backends/nexrad_level2.py b/xradar/io/backends/nexrad_level2.py
@@ -1642,7 +1642,7 @@ def open_nexradlevel2_datatree(
     # todo: refactor _assign_root and _get_subgroup to recieve dict instead of list of datasets.
     # avoiding remove the attributes in the following line
     sweep_dict = {
-        sweep_path: sweep_dict[sweep_path].drop_attrs()
+        sweep_path: sweep_dict[sweep_path].drop_attrs(deep=False)
         for sweep_path in sweep_dict.keys()
     }
     dtree = dtree | sweep_dict