Feature exists in dataset, but still it says not found.

I was trying to build a model where I was taking below model_descriptions.yaml
`# Definition of the entities
entities:
- name: node
state_dimension: 16
initial_state:
- type: build_state
input:
- $id

- name: path
state_dimension: 32
initial_state:
- type: build_state
input:
- $traffic
- $bandwidth
- $latency
`
As my data looks like:
`[{"directed": true, "multigraph": false, "graph": {}, "nodes": [{"entity": "node", "id": 0}, {"entity": "node", "id": 1}, {"entity": "node", "id": 2}, {"entity": "node", "id": 3}, {"entity": "node", "id": 4}, {"entity": "node", "id": 5}, {"entity": "node", "id": 6}, {"entity": "node", "id": 7}], "links": [{"bandwidth": 20000, "traffic": 7.833, "latency": 758.705, "delay": 758.705, "entity": "path", "source": 4, "target": 7}, {"bandwidth": 30000, "traffic": 7.822, "latency": 698.062, "delay": 698.062, "entity": "path", "source": 5, "target": 3}, {"bandwidth": 30000, "traffic": 4.592, "latency": 698.102, "delay": 698.102, "entity": "path", "source": 5, "target": 6}, {"bandwidth": 45000, "traffic": 1.306, "latency": 895.771, "delay": 895.771, "entity": "path", "source": 6, "target": 0}, {"bandwidth": 50000, "traffic": 5.937, "latency": 309.396, "delay": 309.396, "entity": "path", "source": 7, "target": 4}]}]`
Can someone please tell any mistake which I might be doing.