Avoid object cast (#1464)

Co-authored-by: Florian Jetter <[email protected]>
coiled · Apr 18, 2024 · 46f0a3e · 46f0a3e
1 parent a1850d1
commit 46f0a3e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/tpch/dask_queries.py b/tests/tpch/dask_queries.py
@@ -65,8 +65,7 @@ def query_2(dataset_path, fs, scale):
         partsupp_filtered, left_on="s_suppkey", right_on="ps_suppkey", how="inner"
     )
     part_filtered = part_filtered[
-        (part_filtered["p_size"] == var1)
-        & (part_filtered["p_type"].astype(str).str.endswith(var2))
+        (part_filtered["p_size"] == var1) & (part_filtered["p_type"].str.endswith(var2))
     ]
     merged_df = part_filtered.merge(
         ps_s_r_n_merged, left_on="p_partkey", right_on="ps_partkey", how="inner"