Type MSStructureSimulator.transform_data (#40)

ratt-ru · Oct 16, 2024 · 4295680 · 4295680
1 parent c18e83d
commit 4295680
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/doc/source/changelog.rst b/doc/source/changelog.rst
@@ -5,7 +5,7 @@ Changelog
 
 X.Y.Z (DD-MM-YYYY)
 ------------------
-* Add test case for irregular grids (:pr:`39`)
+* Add test case for irregular grids (:pr:`39`, :pr:`40`)
 * Rename MSv2PartitionEntryPoint to MSv2EntryPoint (:pr:`38`)
 * Move ``chunks`` kwarg functionality in MSv2PartitionEntryPoint.open_datatree
   to ``partition_chunks`` (:pr:`37`)

diff --git a/xarray_ms/testing/simulator.py b/xarray_ms/testing/simulator.py
@@ -112,6 +112,7 @@ class MSStructureSimulator:
   simulate_data: bool
   model: Dict[str, Any]
   data_description: DataDescription
+  transform_data: Callable[[PartitionDataType], PartitionDataType] | None
 
   def __init__(
     self,