Fix mypy issue

equinor · Oct 15, 2024 · 1245d95 · 1245d95
1 parent 9daef14
commit 1245d95
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/subscript/check_swatinit/check_swatinit.py b/src/subscript/check_swatinit/check_swatinit.py
@@ -2,7 +2,7 @@
 
 import argparse
 import sys
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Union
 
 import numpy as np
 import pandas as pd
@@ -473,7 +473,7 @@ def _evaluate_pc(
     swats: List[float],
     scale_vert: List[float],
     swls: List[float],
-    swus: List[float],
+    swus: Union[None, List[float]],
     satfunc: pd.DataFrame,
     sat_name: str = "SW",
     pc_name: str = "PCOW",