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

feat: add torch type declaration

Browse source
This commit is contained in:
Shunsuke Shibayama 2023-12-09 12:05:07 +09:00
parent 480c9e4f83
commit 1aa61cd6a6
24 changed files with 84 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

0
crates/erg_compiler/lib/external/torchvision.d/__init__.d.er vendored Normal file
View file

0
crates/erg_compiler/lib/external/torchvision.d/datasets.d/__init__.d.er vendored Normal file
View file

23
crates/erg_compiler/lib/external/torchvision.d/datasets.d/mnist.d.er vendored Normal file
View file

@ -0,0 +1,23 @@
vision = pyimport "./vision"
.MNIST: ClassType
.MNIST <: vision.VisionDataset
.MNIST.
__call__: (
root: Str,
train := Bool,
download := Bool,
transform := GenericCallable,
target_transform := GenericCallable,
) -> .MNIST
.FashionMNIST: ClassType
.FashionMNIST <: .MNIST
.FashionMNIST.
__call__: (
root: Str,
train := Bool,
download := Bool,
transform := GenericCallable,
target_transform := GenericCallable,
) -> .FashionMNIST

0
crates/erg_compiler/lib/external/torchvision.d/datasets.d/utils.d.er vendored Normal file
View file

9
crates/erg_compiler/lib/external/torchvision.d/datasets.d/vision.d.er vendored Normal file
View file

@ -0,0 +1,9 @@
.VisionDataset: ClassType
.VisionDataset.
__call__: (
root: Str,
train := Bool,
download := Bool,
transform := GenericCallable,
target_transform := GenericCallable,
) -> .VisionDataset

0
crates/erg_compiler/lib/external/torchvision.d/package.er vendored Normal file
View file

0
crates/erg_compiler/lib/external/torchvision.d/transforms.d/__init__.d.er vendored Normal file
View file

3
crates/erg_compiler/lib/external/torchvision.d/transforms.d/transforms.d.er vendored Normal file
View file

@ -0,0 +1,3 @@
.ToTensor: ClassType
.ToTensor.
__call__: () -> .ToTensor