Typing

pola-rs · May 31, 2024 · 35fbace · 35fbace
1 parent e2f2e70
commit 35fbace
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py-polars/polars/interchange/utils.py b/py-polars/polars/interchange/utils.py
@@ -72,14 +72,14 @@ def polars_dtype_to_dtype(dtype: PolarsDataType) -> Dtype:
 
 
 def _datetime_to_dtype(dtype: Datetime) -> Dtype:
-    tu = dtype.time_unit[0] if dtype.time_unit is not None else "u"
+    tu = dtype.time_unit[0]
     tz = dtype.time_zone if dtype.time_zone is not None else ""
     arrow_c_type = f"ts{tu}:{tz}"
     return DtypeKind.DATETIME, 64, arrow_c_type, NE
 
 
 def _duration_to_dtype(dtype: Duration) -> Dtype:
-    tu = dtype.time_unit[0] if dtype.time_unit is not None else "u"
+    tu = dtype.time_unit[0]
     arrow_c_type = f"tD{tu}"
     return DtypeKind.DATETIME, 64, arrow_c_type, NE