from __future__ import absolute_import
from relaax.server.parameter_server import parameter_server_base
from relaax.server.common import session

from . import ddpg_model


class ParameterServer(parameter_server_base.ParameterServerBase):
    def init_session(self):
        self.session = session.Session(ddpg_model.SharedParameters())
        self.session.op_initialize()
        self.session.op_init_target_weights()

    def n_step(self):
        return self.session.op_n_step()

    def score(self):
        return self.session.op_score()

    def get_session(self):
        return self.session

[]