Fix type issues

sayerhs · Mar 15, 2024 · 0871566 · 0871566
1 parent 726d2d7
commit 0871566
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/caelus/post/__init__.py b/caelus/post/__init__.py
@@ -14,3 +14,8 @@
 
 from .funcobj import PostProcessing
 from .logs import SolverLog
+
+__all__ = [
+    "PostProcessing",
+    "SolverLog",
+]
diff --git a/caelus/py.typed b/caelus/py.typed
diff --git a/caelus/utils/osutils.py b/caelus/utils/osutils.py
@@ -246,6 +246,6 @@ def split_path(fname):
     return (fdir, base, ext)
 
 
-def path_or_cwd(pdir: Path) -> Path:
+def path_or_cwd(pdir: Path | str | None) -> Path:
     """Sanitize the given path. If None, return current workding directory"""
     return Path(abspath(str(pdir))) if pdir is not None else Path.cwd()