Provide default value for labels in tensor form

LukasZahradnik · Oct 24, 2024 · 1dffc33 · 1dffc33
1 parent 0389866
commit 1dffc33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neuralogic/dataset/tensor.py b/neuralogic/dataset/tensor.py
@@ -48,7 +48,7 @@ def __init__(
         self,
         x: Sequence,
         edge_index: Sequence,
-        y: Union[Sequence, float, int],
+        y: Union[Sequence, float, int] = 0.0,
         edge_attr: Optional[Sequence] = None,
         y_mask: Optional[Sequence] = None,
     ):