fix typo

89c21e32 · rusty1s · a1a492ae · 89c21e32
Commit 89c21e32 authored Nov 04, 2021 by rusty1s
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torch_geometric_autoscale/models/base.py torch_geometric_autoscale/models/base.py +1 -1

No files found.
--- a/torch_geometric_autoscale/models/base.py
+++ b/torch_geometric_autoscale/models/base.py
@@ -108,7 +108,7 @@ class ScalableGNN(torch.nn.Module):
        .. code-block:: python

            batch_size = 5
-            offset = [0, 2, 5]
+            offset = [0, 5]
            count = [2, 3]

        Args: