Skip to content

Commit 2fa2af7

Browse files
committed
chore(predicates): remove casts while keeping mypy clean
1 parent e3f2b57 commit 2fa2af7

File tree

1 file changed

+7
-7
lines changed
  • graphistry/compute/predicates

1 file changed

+7
-7
lines changed

graphistry/compute/predicates/str.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from typing import Optional, Union, cast
1+
from typing import Optional, Union
22

33
import pandas as pd
44

@@ -153,10 +153,10 @@ def __call__(self, s: SeriesT) -> SeriesT:
153153
elif not is_cudf and not self.case:
154154
# pandas tuple with case-insensitive - need workaround
155155
if len(self.pat) == 0:
156-
result = cast(pd.Series, pd.Series([False] * len(s), index=s.index))
156+
result = pd.Series([False] * len(s), index=s.index)
157157
# Preserve NA values when na=None (default)
158158
if self.na is None:
159-
result = cast(pd.Series, result.astype(object))
159+
result = result.astype(object)
160160
result[s.isna()] = None
161161
else:
162162
s_lower = s.str.lower()
@@ -178,7 +178,7 @@ def __call__(self, s: SeriesT) -> SeriesT:
178178
has_na: bool = bool(s.isna().any())
179179
if has_na:
180180
# Convert to object dtype to preserve None values
181-
result_pd = cast(pd.Series, result.to_pandas().astype('object'))
181+
result_pd = result.to_pandas().astype('object')
182182
result_pd[s.to_pandas().isna()] = None
183183
result = cudf.from_pandas(result_pd)
184184
else:
@@ -321,10 +321,10 @@ def __call__(self, s: SeriesT) -> SeriesT:
321321
# pandas tuple with case-insensitive - need workaround
322322
if len(self.pat) == 0:
323323
# Create False for all values
324-
result = cast(pd.Series, pd.Series([False] * len(s), index=s.index))
324+
result = pd.Series([False] * len(s), index=s.index)
325325
# Preserve NA values when na=None (default)
326326
if self.na is None:
327-
result = cast(pd.Series, result.astype(object))
327+
result = result.astype(object)
328328
result[s.isna()] = None
329329
else:
330330
s_lower = s.str.lower()
@@ -346,7 +346,7 @@ def __call__(self, s: SeriesT) -> SeriesT:
346346
has_na: bool = bool(s.isna().any())
347347
if has_na:
348348
# Convert to object dtype to preserve None values
349-
result_pd = cast(pd.Series, result.to_pandas().astype('object'))
349+
result_pd = result.to_pandas().astype('object')
350350
result_pd[s.to_pandas().isna()] = None
351351
result = cudf.from_pandas(result_pd)
352352
else:

0 commit comments

Comments
 (0)