Update pages/15_Graphs.py

pages/15_Graphs.py

@@ -23,7 +23,7 @@ def model_fn(graph_tensor_spec: tfgnn.GraphTensorSpec):
         graph = mt_albis.MtAlbisGraphUpdate(
             units=128,
             message_dim=64,
-            receiver_tag=tfgnn.NODES,  # Add this line
+            receiver_tag=tfgnn.NODES,  # Correctly use tfgnn.NODES
             attention_type="none",
             simple_conv_reduce_type="mean",
             normalization_type="layer",
@@ -56,8 +56,8 @@ def create_sample_graph():
             "cites": tfgnn.EdgeSet.from_fields(
                 sizes=[num_edges],
                 adjacency=tfgnn.Adjacency.from_indices(
-                    source=("papers", tf.cast([e[0] for e in graph.edges()], tf.int32)),
-                    target=("papers", tf.cast([e[1] for e in graph.edges()], tf.int32))
+                    source=("papers", tf.constant([e[0] for e in graph.edges()], dtype=tf.int32)),
+                    target=("papers", tf.constant([e[1] for e in graph.edges()], dtype=tf.int32))
                 ),
                 features={"features": edge_features}
             )
@@ -97,4 +97,4 @@ def main():
     st.text(f"Edge feature shape: {graph_tensor.edge_sets['cites']['features'].shape}")
 
 if __name__ == "__main__":
-    main()
+    main()