Skip to content

Commit 4d8c569

Browse files
committed
fix: get_weight for WideSampler
1 parent 39757b9 commit 4d8c569

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

pyrdf2vec/samplers/wide.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,9 +85,7 @@ def get_weight(self, hop: Hop) -> float:
8585
without the sampling strategy having been trained.
8686
8787
"""
88-
if not (
89-
self._pred_degs or self._obj_degs or not self._neighbor_counts
90-
):
88+
if not (self._pred_degs or self._obj_degs or self._neighbor_counts):
9189
raise ValueError(
9290
"You must call the `fit(kg)` method before get the weight of"
9391
+ " a hop."

0 commit comments

Comments
 (0)