KeyError: 'P' Error happens

[LimHyungTae]

I tried to visualize dsg.json file, but once I run notebooks/bindings_demo.py, the below error occurs:

Traceback (most recent call last):
  File "bindings_demo.py", line 68, in <module>
    fig = dsg.plot_scene_graph(G, layer_settings=layer_settings)
  File "/home/shapelim/dcist_ws/src/DCIST-Spark-Core/spark_dsg/python/src/spark_dsg/visualization.py", line 108, in plot_scene_graph
    _draw_layer_nodes(fig, layer, **settings)
  File "/home/shapelim/dcist_ws/src/DCIST-Spark-Core/spark_dsg/python/src/spark_dsg/visualization.py", line 67, in _draw_layer_nodes
    pos.append(np.squeeze(z_offset(node)))
  File "/home/shapelim/dcist_ws/src/DCIST-Spark-Core/spark_dsg/python/src/spark_dsg/visualization.py", line 49, in z_offset
    offset[2] += NODE_TYPE_OFFSET[node.id.category]
KeyError: 'P'

So, I circumvent this issue like that:

if node.id.category == "P":
      offset[2] += NODE_TYPE_OFFSET["p"]
  else :
      offset[2] += NODE_TYPE_OFFSET[node.id.category]