Skip to content

Commit b06622d

Browse files
Update benchmarks.py
1 parent 0223006 commit b06622d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

manify/utils/benchmarks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -416,7 +416,7 @@ def benchmark(
416416
kappa_mlp = KappaGCN(pm=pm_stereo, **nn_kwargs).to(device)
417417
t1 = time.time()
418418
if task == "link_prediction":
419-
kappa_mlp.fit(X_train_stereo, y_train, A=A_train, tqdm_prefix="kappa_mlp", lp_indices= **nn_train_kwargs)
419+
kappa_mlp.fit(X_train_stereo, y_train, A=A_train, tqdm_prefix="kappa_mlp", **nn_train_kwargs)
420420
else:
421421
kappa_mlp.fit(X_train_stereo, y_train, A=None, tqdm_prefix="kappa_mlp", **nn_train_kwargs)
422422
t2 = time.time()

0 commit comments

Comments
 (0)