language-servers/erg
1
0
Fork 0
mirror of https://github.com/erg-lang/erg.git synced 2025-10-02 21:44:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: torch type decls

Browse source
This commit is contained in:
Shunsuke Shibayama 2024-02-03 20:51:58 +09:00
parent 9c9f8b7a0a
commit 3be5d75d05
13 changed files with 149 additions and 83 deletions
Download patch file Download diff file Expand all files Collapse all files

2
crates/erg_compiler/lib/external/torch.d/utils.d/data.d/dataloader.d.er vendored
View file

@ -3,7 +3,7 @@ dataset = pyimport "./dataset"
{Sampler;} = pyimport "./sampler"
.DataLoader: ClassType
.DataLoader <: Iterable((torch.Tensor(_, _), torch.Tensor(_, _)))
.DataLoader <: Iterable((torch.Tensor!(_, _), torch.Tensor!(_, _)))
.DataLoader.
__call__: (
dataset: dataset.Dataset,