Update strptime() docstrings and type hints to include Time (#3502)

pola-rs · May 26, 2022 · afb4df7 · afb4df7
1 parent 1fb0ae3
commit afb4df7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/py-polars/polars/internals/expr.py b/py-polars/polars/internals/expr.py
@@ -3656,18 +3656,18 @@ def __init__(self, expr: Expr):
 
     def strptime(
         self,
-        datatype: Union[Type[Date], Type[Datetime]],
+        datatype: Union[Type[Date], Type[Datetime], Type[Time]],
         fmt: Optional[str] = None,
         strict: bool = True,
         exact: bool = True,
     ) -> Expr:
         """
-        Parse utf8 expression as a Date/Datetimetype.
+        Parse a UTF8 expression to a Date/Datetime/Time type.
 
         Parameters
         ----------
         datatype
-            Date | Datetime.
+            Date | Datetime | Time.
         fmt
             format to use, see the following link for examples:
             https://docs.rs/chrono/latest/chrono/format/strftime/index.html
@@ -3732,7 +3732,7 @@ def strptime(
             return wrap_expr(self._pyexpr.str_parse_time(fmt, strict, exact))
         else:
             raise ValueError(
-                "dtype should be of type {Date, Datetime}"
+                "dtype should be of type {Date, Datetime, Time}"
             )  # pragma: no cover
 
     def lengths(self) -> Expr:

diff --git a/py-polars/polars/internals/series.py b/py-polars/polars/internals/series.py
@@ -3754,7 +3754,7 @@ def __init__(self, series: "Series"):
 
     def strptime(
         self,
-        datatype: Union[Type[Date], Type[Datetime]],
+        datatype: Union[Type[Date], Type[Datetime], Type[Time]],
         fmt: Optional[str] = None,
         strict: bool = True,
         exact: bool = True,
@@ -3765,7 +3765,7 @@ def strptime(
         Parameters
         ----------
         datatype
-            Date or Datetime.
+            Date, Datetime or Time.
         fmt
             format to use, see the following link for examples:
             https://docs.rs/chrono/latest/chrono/format/strftime/index.html
@@ -3779,7 +3779,7 @@ def strptime(
 
         Returns
         -------
-        A Date/ Datetime Series
+        A Date / Datetime / Time Series
 
         Examples
         --------