linting

6146660b · rusty1s · b5ac9f33 · 6146660b · 6146660b
Commit 6146660b authored Apr 10, 2018 by rusty1s
Show whitespace changes
Inline Side-by-side

Showing with 8 additions and 5 deletions

test/test_conv.py test/test_conv.py +4 -4

torch_spline_conv/conv.py torch_spline_conv/conv.py +4 -1

No files found.
--- a/test/test_conv.py
+++ b/test/test_conv.py
--- a/torch_spline_conv/conv.py
+++ b/torch_spline_conv/conv.py
+# import torch
 def spline_conv(x,
                edge_index,
                pseudo,
@@ -7,4 +10,4 @@ def spline_conv(x,
                degree=1,
                root_weight=None,
                bias=None):
-    pass
+    pseudo = pseudo.unsqueeze(-1) if pseudo.dim() == 1 else pseudo