From 6663f6744784badee84a83197dc00af2afc884e1 Mon Sep 17 00:00:00 2001
From: Simon Perkins <simon.perkins@gmail.com>
Date: Wed, 16 Oct 2024 15:04:04 +0200
Subject: [PATCH] Type MSStructureSimulator.transform_data

---
 xarray_ms/testing/simulator.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/xarray_ms/testing/simulator.py b/xarray_ms/testing/simulator.py
index 8d4f821..b133777 100644
--- 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,