00536cb4777442076c1fb6c3026f50d72b9c0247,cistar_dev/cistar_dev/visualizer_CISTAR_tf.py,,,#,18

Before Change

        initial_config = scenario.initial_config
        if TYPE == "figure8":
            scenario = Figure8Scenario(exp_tag, type_params, net_params, cfg_params, initial_config=initial_config)
        elif TYPE == "loop":
            scenario = LoopScenario(exp_tag, type_params, net_params, cfg_params, initial_config=initial_config)
        env._wrapped_env.scenario = scenario

After Change

        data = joblib.load(args.file)
        policy = data["policy"]
        env = data["env"]
        import ipdb; ipdb.set_trace()

        // Input
        unwrapped_env = env.env.unwrapped
        if unwrapped_env.obs_var_labels:

Instances