844a5af831a66505f3ee92dbea51f052a4f5262d,rl_coach/rollout_worker.py,,main,#,73

Before Change

    if args.memory_backend_params:
        args.memory_backend_params = json.loads(args.memory_backend_params)
        if "run_type" not in args.memory_backend_params:
            args.memory_backend_params["run_type"] = "worker"
        graph_manager.agent_params.memory.register_var("memory_backend_params", construct_memory_params(args.memory_backend_params))
    rollout_worker(
        graph_manager=graph_manager,
        checkpoint_dir=args.checkpoint_dir,

After Change

        graph_manager.agent_params.memory.register_var("memory_backend_params", construct_memory_params(args.memory_backend_params))

    if args.data_store_params:
        data_store_params = construct_data_store_params(json.loads(args.data_store_params))
        data_store_params.checkpoint_dir = args.checkpoint_dir
        graph_manager.data_store_params = data_store_params
        data_store = get_data_store(data_store_params)
        wait_for_checkpoint(checkpoint_dir=args.checkpoint_dir, data_store=data_store)
        // thread = Thread(target = data_store_ckpt_load, args = [data_store])
        // thread.start()

Instances