Fix #46

[Subhiiiiii]

Fix: Handle None senders/receivers in GCN when add_self_edges=True

Problem

The GCN implementation would fail when:

1. The graph had no edges (senders and receivers are None)
2. add_self_edges=True was set

This occurred because jnp.concatenate() was called with None values:

conv_receivers = jnp.concatenate((receivers, jnp.arange(total_num_nodes)), axis=0)  # fails when receivers is None

Solution

Added explicit handling of None values by converting them to empty arrays:

if senders is None:
    senders = jnp.array([], dtype=jnp.int32)
if receivers is None:
    receivers = jnp.array([], dtype=jnp.int32)

Impact

• Fixes: Enables GCN to work with edgeless graphs when add_self_edges=True
• Consistency: Maintains same behavior for graphs with/without edges
• Robustness: More type-safe implementation
• Backward Compatible: No changes to existing functionality for graphs with edges

Example

For a 3-node graph with no edges:

# Before (would crash):
senders = None
receivers = None

# After (works correctly):
senders = jnp.array([], dtype=jnp.int32)  # shape (0,)
receivers = jnp.array([], dtype=jnp.int32)  # shape (0,)