Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fall back when casting a timestamp to numeric in cudf-polars #16232

Merged
merged 8 commits into from
Jul 23, 2024
Merged

Fall back when casting a timestamp to numeric in cudf-polars #16232

Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
f54dd43
raise when casting from timestamp to numeric
brandon-b-miller Jul 9, 2024
2f6a6ff
Merge branch 'branch-24.08' into fix-cudf-polars-timestamp-casting
brandon-b-miller Jul 9, 2024
1c6db1b
use libcudf traits
brandon-b-miller Jul 9, 2024
35c0e70
cleanup
brandon-b-miller Jul 9, 2024
1db84d6
Merge branch 'branch-24.08' into fix-cudf-polars-timestamp-casting
brandon-b-miller Jul 22, 2024
dc0ebbb
use is_supported_cast
brandon-b-miller Jul 22, 2024
a140cd0
Update python/cudf_polars/cudf_polars/dsl/expr.py
brandon-b-miller Jul 22, 2024
a2858e6
basic tests
brandon-b-miller Jul 22, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
use libcudf traits
  • Loading branch information
@brandon-b-miller
brandon-b-miller committed Jul 9, 2024
commit 1c6db1bc0db7169a1ad33babd2884c73255794dc
19 changes: 1 addition & 18 deletions python/cudf_polars/cudf_polars/dsl/expr.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1123,24 +1123,7 @@ def __init__(self, dtype: plc.DataType, value: Expr) -> None:
"Need to implement cast to/from string separately."
)
# TODO: use exposed libcudf trait checking APIs
elif self.dtype.id() in {
plc.TypeId.TIMESTAMP_DAYS,
plc.TypeId.TIMESTAMP_MICROSECONDS,
plc.TypeId.TIMESTAMP_MILLISECONDS,
plc.TypeId.TIMESTAMP_NANOSECONDS,
} and value.dtype.id() in {
plc.TypeId.INT8,
plc.TypeId.INT16,
plc.TypeId.INT32,
plc.TypeId.INT64,
plc.TypeId.UINT8,
plc.TypeId.UINT16,
plc.TypeId.UINT32,
plc.TypeId.UINT64,
plc.TypeId.FLOAT32,
plc.TypeId.FLOAT64,
plc.TypeId.BOOL,
}:
elif plc.traits.is_chrono(self.dtype) and plc.traits.is_numeric(value.dtype):
raise NotImplementedError("Can't cast duration to numeric")

def do_evaluate(
Expand Down
Loading