From 03f0887a2d7175cacf716f76e1bd0716040072be Mon Sep 17 00:00:00 2001
From: Marshall Crumiller <mcrumiller@gmail.com>
Date: Thu, 16 Jan 2025 09:13:12 -0500
Subject: [PATCH] Assume correct schema in impl

---
 .../src/dsl/function_expr/range/linear_space.rs     | 13 ++-----------
 1 file changed, 2 insertions(+), 11 deletions(-)

diff --git a/crates/polars-plan/src/dsl/function_expr/range/linear_space.rs b/crates/polars-plan/src/dsl/function_expr/range/linear_space.rs
index 0e46e43261a2..647b1ff0cfe5 100644
--- a/crates/polars-plan/src/dsl/function_expr/range/linear_space.rs
+++ b/crates/polars-plan/src/dsl/function_expr/range/linear_space.rs
@@ -37,7 +37,7 @@ pub(super) fn linear_space(s: &[Column], closed: ClosedInterval) -> PolarsResult
             closed,
             name.clone(),
         ),
-        (mut dt, dt2) if dt.is_temporal() && dt == dt2 => {
+        (mut dt, _) if dt.is_temporal() => {
             let mut start = start.extract::<i64>().unwrap();
             let mut end = end.extract::<i64>().unwrap();
 
@@ -50,16 +50,7 @@ pub(super) fn linear_space(s: &[Column], closed: ClosedInterval) -> PolarsResult
             new_linear_space_f64(start as f64, end as f64, num_samples, closed, name.clone())
                 .map(|s| s.cast(&dt).unwrap())
         },
-        (dt1, dt2) if !dt1.is_primitive_numeric() || !dt2.is_primitive_numeric() => {
-            Err(PolarsError::ComputeError(
-                format!(
-                    "'start' and 'end' have incompatible dtypes, got {:?} and {:?}",
-                    dt1, dt2
-                )
-                .into(),
-            ))
-        },
-        (_, _) => new_linear_space_f64(
+        _ => new_linear_space_f64(
             start.extract::<f64>().unwrap(),
             end.extract::<f64>().unwrap(),
             num_samples,