Fix the typing annotation for the return of parse() (#452)

python-pendulum · Jul 13, 2020 · 1b1784f · 1b1784f
1 parent 8af1e2e
commit 1b1784f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pendulum/parser.py b/pendulum/parser.py
@@ -5,8 +5,11 @@
 
 import pendulum
 
+from .date import Date
 from .parsing import _Interval
 from .parsing import parse as base_parse
+from .time import Duration
+from .time import Time
 from .tz import UTC
 
 
@@ -16,7 +19,9 @@
     CDuration = None
 
 
-def parse(text, **options):  # type: (str, **typing.Any) -> str
+def parse(
+    text, **options
+):  # type: (str, **typing.Any) -> typing.Union[Date, Time, Duration]
     # Use the mock now value if it exists
     options["now"] = options.get("now", pendulum.get_test_now())