Skip to content

Commit

Permalink
Merge branch 'join-where' of https://github.com/pola-rs/r-polars into…
Browse files Browse the repository at this point in the history
… join-where
  • Loading branch information
etiennebacher committed Oct 3, 2024
2 parents 8f54eea + bf6f820 commit 71b8274
Show file tree
Hide file tree
Showing 13 changed files with 142 additions and 40 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/flake.yml
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ on:
- cron: '0 0 * * 0' # runs weekly on Sunday at 00:00
permissions:
contents: write
pull-requests: write

jobs:
lockfile:
Expand All @@ -17,5 +18,4 @@ jobs:
- name: Update flake.lock
uses: DeterminateSystems/update-flake-lock@v24
with:
token: ${{ secrets.GITHUB_TOKEN }}
pr-title: "chore: update flake.lock"
1 change: 1 addition & 0 deletions NEWS.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
- New argument `strict` in `$drop()` to determine whether unknown column names
should trigger an error (#1220).
- New method `$to_dummies()` for `DataFrame` (#1225).
- New argument `include_file_paths` in `pl_scan_csv()` and `pl_read_csv()` (#1235).
- New method `$join_where()` for `DataFrame` and `LazyFrame` to perform
inequality joins (#1237).

Expand Down
2 changes: 1 addition & 1 deletion R/extendr-wrappers.R
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ concat_df_horizontal <- function(l) .Call(wrap__concat_df_horizontal, l)

concat_series <- function(l, rechunk, to_supertypes) .Call(wrap__concat_series, l, rechunk, to_supertypes)

new_from_csv <- function(path, has_header, separator, comment_prefix, quote_char, skip_rows, dtypes, null_values, ignore_errors, cache, infer_schema_length, n_rows, encoding, low_memory, rechunk, skip_rows_after_header, row_index_name, row_index_offset, try_parse_dates, eol_char, raise_if_empty, truncate_ragged_lines) .Call(wrap__new_from_csv, path, has_header, separator, comment_prefix, quote_char, skip_rows, dtypes, null_values, ignore_errors, cache, infer_schema_length, n_rows, encoding, low_memory, rechunk, skip_rows_after_header, row_index_name, row_index_offset, try_parse_dates, eol_char, raise_if_empty, truncate_ragged_lines)
new_from_csv <- function(path, has_header, separator, comment_prefix, quote_char, skip_rows, dtypes, null_values, ignore_errors, cache, infer_schema_length, n_rows, encoding, low_memory, rechunk, skip_rows_after_header, row_index_name, row_index_offset, try_parse_dates, eol_char, raise_if_empty, truncate_ragged_lines, include_file_paths) .Call(wrap__new_from_csv, path, has_header, separator, comment_prefix, quote_char, skip_rows, dtypes, null_values, ignore_errors, cache, infer_schema_length, n_rows, encoding, low_memory, rechunk, skip_rows_after_header, row_index_name, row_index_offset, try_parse_dates, eol_char, raise_if_empty, truncate_ragged_lines, include_file_paths)

import_arrow_ipc <- function(path, n_rows, cache, rechunk, row_name, row_index, hive_partitioning, hive_schema, try_parse_hive_dates, include_file_paths) .Call(wrap__import_arrow_ipc, path, n_rows, cache, rechunk, row_name, row_index, hive_partitioning, hive_schema, try_parse_hive_dates, include_file_paths)

Expand Down
8 changes: 6 additions & 2 deletions R/io_csv.R
Original file line number Diff line number Diff line change
Expand Up @@ -65,6 +65,8 @@
#' @param truncate_ragged_lines Truncate lines that are longer than the schema.
#' @param reuse_downloaded If `TRUE`(default) and a URL was provided, cache the
#' downloaded files in session for an easy reuse.
#' @param include_file_paths Include the path of the source file(s) as a column
#' with this name.
#' @return [LazyFrame][LazyFrame_class]
#' @examples
#' my_file = tempfile()
Expand Down Expand Up @@ -97,7 +99,8 @@ pl_scan_csv = function(
eol_char = "\n",
raise_if_empty = TRUE,
truncate_ragged_lines = FALSE,
reuse_downloaded = TRUE) {
reuse_downloaded = TRUE,
include_file_paths = NULL) {
# capture all args and modify some to match lower level function
args = as.list(environment())

Expand Down Expand Up @@ -181,7 +184,8 @@ pl_read_csv = function(
eol_char = "\n",
raise_if_empty = TRUE,
truncate_ragged_lines = FALSE,
reuse_downloaded = TRUE) {
reuse_downloaded = TRUE,
include_file_paths = NULL) {
.args = as.list(environment())
result({
do.call(pl$scan_csv, .args)$collect()
Expand Down
30 changes: 15 additions & 15 deletions flake.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 5 additions & 1 deletion man/IO_read_csv.Rd

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions man/IO_read_parquet.Rd

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 5 additions & 1 deletion man/IO_scan_csv.Rd

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

4 changes: 2 additions & 2 deletions man/IO_scan_parquet.Rd

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

97 changes: 85 additions & 12 deletions src/rust/Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 3 additions & 3 deletions src/rust/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -42,14 +42,14 @@ extendr-api = { git = "https://github.com/extendr/extendr", rev = "1895bfc8ee223
] }
flume = "0.11.0"
indenter = "0.3.3"
ipc-channel = "0.18.1"
once_cell = "1.20.0"
ipc-channel = "0.18.3"
once_cell = "1.20.1"
rayon = "1.10.0"
serde = { version = "1.0.210", features = ["derive"] }
serde_json = "*"
smartstring = "1.0.1"
state = "0.6.0"
thiserror = "1.0.63"
thiserror = "1.0.64"
polars-core = { git = "https://github.com/pola-rs/polars.git", rev = "54218e7e35e3defd4b0801e820c56eea6b91e525", default-features = false }
polars-lazy = { git = "https://github.com/pola-rs/polars.git", rev = "54218e7e35e3defd4b0801e820c56eea6b91e525", default-features = false }
either = "1"
Expand Down
Loading

0 comments on commit 71b8274

Please sign in to comment.