diff --git a/graphlearn_torch/python/sampler/base.py b/graphlearn_torch/python/sampler/base.py index 5b831b3b..23eaf1f9 100644 --- a/graphlearn_torch/python/sampler/base.py +++ b/graphlearn_torch/python/sampler/base.py @@ -349,7 +349,7 @@ class SamplingConfig: with_neg: bool with_weight: bool edge_dir: Literal['in', 'out'] - seed: int + seed: Optional[int] class BaseSampler(ABC):