From c63ad8ac92e5bcf488e3b863c188dcc473cd5a87 Mon Sep 17 00:00:00 2001
From: Umut <umutsahin@protonmail.com>
Date: Sat, 7 Aug 2021 23:45:12 +0300
Subject: [PATCH] fix(debugging): assign a color to ArbitraryFunction nodes

---
 hdk/common/debugging/draw_graph.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/hdk/common/debugging/draw_graph.py b/hdk/common/debugging/draw_graph.py
index 33cf49bd7..8454dc5f4 100644
--- a/hdk/common/debugging/draw_graph.py
+++ b/hdk/common/debugging/draw_graph.py
@@ -13,6 +13,7 @@ IR_NODE_COLOR_MAPPING = {
     ir.Add: "red",
     ir.Sub: "yellow",
     ir.Mul: "green",
+    ir.ArbitraryFunction: "orange",
     "output": "magenta",
 }