From 95eb19b8da9b4646c086a83ffb9633d52d2ce3a5 Mon Sep 17 00:00:00 2001
From: Kyle Montemayor <kyle.e.montemayor@gmail.com>
Date: Wed, 30 Apr 2025 10:28:07 -0700
Subject: [PATCH] Annotate SamplingConfig.seed as optional

---
 graphlearn_torch/python/sampler/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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):