fix: overload resolution bug

This commit is contained in:
Shunsuke Shibayama 2024-02-09 14:33:58 +09:00
parent 66352ddd3a
commit 21d5f22ca8
5 changed files with 30 additions and 1 deletions

View file

@ -1,7 +1,10 @@
dataset = pyimport "torch/utils/data/dataset"
{Tensor!;} = pyimport "torch"
.VisionDataset: ClassType
.VisionDataset <: dataset.Dataset
.VisionDataset|<: Indexable(Nat, (Tensor!(Float, _), Tensor!(Float, _)))|.
__getitem__: (index: Nat) -> (Tensor!(Float, _), Tensor!(Float, _))
.VisionDataset.
__call__: (
root: Str,