[TASK]: collapse observable references for stratified model diagram

[shawnyama]

Describe the task Apply a collapse to the observable summary's references so that base states are referred to instead of the actual states when viewing a stratified model.