make .obsmap/.varmap writeable again

closes #138

Author: ilia-kats

CHANGELOG.md

@@ -8,6 +8,12 @@ and this project adheres to [Semantic Versioning][].
 [keep a changelog]: https://keepachangelog.com/en/1.1.0/
 [semantic versioning]: https://semver.org/spec/v2.0.0.html
 
+## [0.3.6]
+
+### Fixed
+
+- `.obsmap` and `.varmap` are once again writeable
+
 ## [0.3.5]
 
 ### Fixed
@@ -150,6 +156,7 @@ To copy the annotations explicitly, you will need to use `pull_obs()` and/or `pu
 
 Initial `mudata` release with `MuData`, previously a part of the `muon` framework.
 
+[0.3.6]: https://github.com/scverse/mudata/releases/tag/v0.3.6
 [0.3.5]: https://github.com/scverse/mudata/releases/tag/v0.3.5
 [0.3.4]: https://github.com/scverse/mudata/releases/tag/v0.3.4
 [0.3.3]: https://github.com/scverse/mudata/releases/tag/v0.3.3

src/mudata/_core/mudata.py

@@ -1482,7 +1482,7 @@ def obsmap(self) -> Mapping[str]:
         in the array contains the numerical index of the observation in the respective modality corresponding to the :class:`MuData`
         observation in that position. The index is 1-based, 0 indicates that the observation is missing in the modality.
         """
-        return MappingProxyType(self._obsmap)
+        return self._obsmap
 
     @property
     def varm(self) -> MutableMapping[str]:
@@ -1532,7 +1532,7 @@ def varmap(self) -> Mapping[str]:
         in the array contains the numerical index of the feature in the respective modality corresponding to the :class:`MuData`
         feature in that position. The index is 1-based, 0 indicates that the feature is missing in the modality.
         """
-        return MappingProxyType(self._varmap)
+        return self._varmap
 
     # Unstructured annotations
 

tests/test_obs_var.py

@@ -44,6 +44,10 @@ def test_obs_vector(mdata: md.MuData):
         mdata.obs_vector("foo")
 
 
+def test_obsmap_writeable(mdata: md.MuData):
+    mdata.obsmap["test_writeable"] = np.arange(mdata.n_obs)
+
+
 @pytest.mark.parametrize("mdata", (0, 1), indirect=True)
 def test_var_global_columns(mdata: md.MuData, filepath_h5mu: str | Path):
     mdata.var.drop(columns=mdata.var.columns, inplace=True)
@@ -89,6 +93,10 @@ def test_var_vector(rng: np.random.Generator, mdata: md.MuData):
         mdata.var_vector("assert-boolean-1")
 
 
+def test_varmap_writeable(mdata: md.MuData):
+    mdata.varmap["test_writeable"] = np.arange(mdata.n_vars)
+
+
 @pytest.mark.parametrize("mdata", (0, 1), indirect=True)
 def test_names_make_unique(mdata: md.MuData):
     attr = "obs" if mdata.axis == 0 else "var"