More colors.

lynxkite-app/web/src/workspace/nodes/LynxKiteNode.tsx

@@ -52,9 +52,11 @@ function getHandles(inputs: any[], outputs: any[]) {
 }
 
 const OP_COLORS: { [key: string]: string } = {
+  pink: "oklch(75% 0.2 0)",
   orange: "oklch(75% 0.2 55)",
+  green: "oklch(75% 0.2 150)",
   blue: "oklch(75% 0.2 230)",
-  green: "oklch(75% 0.2 130)",
+  purple: "oklch(75% 0.2 290)",
 };
 
 function LynxKiteNodeComponent(props: LynxKiteNodeProps) {

lynxkite-graph-analytics/src/lynxkite_graph_analytics/pytorch/pytorch_ops.py

@@ -8,7 +8,7 @@ from .pytorch_core import op, reg, ENV
 
 reg("Input: tensor", outputs=["output"], params=[P.basic("name")])
 reg("Input: graph edges", outputs=["edges"])
-reg("Input: sequential", outputs=["y"])
+reg("Input: sequential", outputs=["y"], params=[P.basic("name")])
 reg("Output", inputs=["x"], outputs=["x"], params=[P.basic("name")])
 
 
@@ -19,6 +19,7 @@ def lstm(x, *, input_size=1024, hidden_size=1024, dropout=0.0):
 
 reg(
     "Neural ODE",
+    color="blue",
     inputs=["x"],
     params=[
         P.basic("relative_tolerance"),
@@ -99,6 +100,7 @@ def concatenate(a, b):
 
 reg(
     "Graph conv",
+    color="blue",
     inputs=["x", "edges"],
     outputs=["x"],
     params=[P.options("type", ["GCNConv", "GATConv", "GATv2Conv", "SAGEConv"])],