File tree Expand file tree Collapse file tree 2 files changed +6
-3
lines changed Expand file tree Collapse file tree 2 files changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -307,6 +307,7 @@ def to_pyarrow_batches(
307
307
def to_torch (
308
308
self ,
309
309
expr : ir .Expr ,
310
+ / ,
310
311
* ,
311
312
params : Mapping [ir .Scalar , Any ] | None = None ,
312
313
limit : int | str | None = None ,
Original file line number Diff line number Diff line change @@ -1357,10 +1357,11 @@ def execute(
1357
1357
def to_torch (
1358
1358
self ,
1359
1359
expr : ir .Expr ,
1360
+ / ,
1360
1361
* ,
1361
1362
params : Mapping [ir .Scalar , Any ] | None = None ,
1362
1363
limit : int | str | None = None ,
1363
- ** _ : Any ,
1364
+ ** kwargs : Any ,
1364
1365
) -> dict [str , torch .Tensor ]:
1365
1366
"""Execute an expression and return results as a dictionary of torch tensors.
1366
1367
@@ -1379,9 +1380,10 @@ def to_torch(
1379
1380
-------
1380
1381
dict[str, torch.Tensor]
1381
1382
A dictionary of torch tensors, keyed by column name.
1382
-
1383
1383
"""
1384
- return self ._to_duckdb_relation (expr , params = params , limit = limit ).torch ()
1384
+ return self ._to_duckdb_relation (
1385
+ expr , params = params , limit = limit , ** kwargs
1386
+ ).torch ()
1385
1387
1386
1388
@util .experimental
1387
1389
def to_parquet (
You can’t perform that action at this time.
0 commit comments