didn't account for None

microsoft · Feb 23, 2024 · f8d42f2 · f8d42f2
1 parent eb65c25
commit f8d42f2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torchgeo/trainers/base.py b/torchgeo/trainers/base.py
@@ -35,7 +35,9 @@ def __init__(self, ignore: Optional[Union[Sequence[str], str]] = None) -> None:
             ignore: Arguments to skip when saving hyperparameters.
         """
         super().__init__()
-        if isinstance(ignore, str):
+        if ignore is None:
+            ignore = ["ignore"]
+        elif isinstance(ignore, str):
             ignore = [ignore, "ignore"]
         else:
             ignore = list(ignore) + ["ignore"]