patch

Signed-off-by: Emile Tenezakis <e.tenezakis@gmail.com>
os-climate · Apr 19, 2024 · 2a3bb40 · 2a3bb40
1 parent 4bb66b6
commit 2a3bb40
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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):