Implement state loading for the `behavior_test_scene_name` in a less sketchy way

[NishanthJKumar]

Currently, we load the test scene if the task num is > 10 and load the train scene otherwise. This is a bit sketchy (what if we decide to change the train/test split ever?). We should do the loading in a better, less sketchy way.