From 2a3bb4023fa18e22e9dbc2e5b50006f4e735ad54 Mon Sep 17 00:00:00 2001
From: Emile Tenezakis <e.tenezakis@gmail.com>
Date: Fri, 19 Apr 2024 10:30:17 +0000
Subject: [PATCH] patch

Signed-off-by: Emile Tenezakis <e.tenezakis@gmail.com>
---
 tests/conftest.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index 0a659d8..ea1cdc6 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,6 +1,6 @@
 import os
 from datetime import datetime
-from typing import Dict, Tuple
+from typing import Dict, Optional, Tuple
 
 import numpy as np
 import pandas as pd  # type: ignore
@@ -47,7 +47,7 @@ class TestTarget(WriteDataset):
     def __init__(self):
         self.datasets = {}
 
-    def write(self, path: str, dataset: xr.Dataset):
+    def write(self, path: str, dataset: xr.Dataset, spatial_coords: Optional[bool] = False):
         self.datasets[path] = dataset
 
     def read(self, path: str):