Skip to content

Commit 6befe11

Browse files
committed
update temporal comparison argument
1 parent f60a200 commit 6befe11

File tree

2 files changed

+10
-6
lines changed

2 files changed

+10
-6
lines changed

python/cugraph-pyg/cugraph_pyg/loader/link_neighbor_loader.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -129,8 +129,10 @@ def __init__(
129129
all workers. If not provided, it will be automatically
130130
calculated.
131131
See cugraph_pyg.sampler.BaseDistributedSampler.
132-
temporal_comparison: str (optional, default='<=', i.e. monotonically decreasing)
133-
The comparison operator for temporal sampling (>, <, >=, <=, last).
132+
temporal_comparison: str (optional, default='monotonically decreasing')
133+
The comparison operator for temporal sampling
134+
('strictly increasing', 'monotonically increasing',
135+
'strictly decreasing', 'monotonically decreasing', 'last').
134136
Note that this should be 'last' for temporal_strategy='last'.
135137
See cugraph_pyg.sampler.BaseDistributedSampler.
136138
**kwargs
@@ -140,7 +142,7 @@ def __init__(
140142
subgraph_type = torch_geometric.sampler.base.SubgraphType(subgraph_type)
141143

142144
if temporal_comparison is None:
143-
temporal_comparison = "<="
145+
temporal_comparison = "monotonically decreasing"
144146

145147
if not directed:
146148
subgraph_type = torch_geometric.sampler.base.SubgraphType.induced

python/cugraph-pyg/cugraph_pyg/loader/neighbor_loader.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -122,8 +122,10 @@ def __init__(
122122
all workers. If not provided, it will be automatically
123123
calculated.
124124
See cugraph_pyg.sampler.BaseDistributedSampler.
125-
temporal_comparison: str (optional, default='<=', i.e. monotonically decreasing)
126-
The comparison operator for temporal sampling (>, <, >=, <=, last).
125+
temporal_comparison: str (optional, default='monotonically decreasing')
126+
The comparison operator for temporal sampling
127+
('strictly increasing', 'monotonically increasing',
128+
'strictly decreasing', 'monotonically decreasing', 'last').
127129
Note that this should be 'last' for temporal_strategy='last'.
128130
See cugraph_pyg.sampler.BaseDistributedSampler.
129131
**kwargs
@@ -133,7 +135,7 @@ def __init__(
133135
subgraph_type = torch_geometric.sampler.base.SubgraphType(subgraph_type)
134136

135137
if temporal_comparison is None:
136-
temporal_comparison = "<="
138+
temporal_comparison = "monotonically decreasing"
137139

138140
if not directed:
139141
subgraph_type = torch_geometric.sampler.base.SubgraphType.induced

0 commit comments

Comments
 (0)