377e7de90e0ab1e973d9bed4909d26abab68e7e2,slm_lab/agent/algorithm/base.py,Algorithm,train,#Algorithm#,89

Before Change

    @lab_api
    def train(self):
        """Implement algorithm train, or throw NotImplementedError"""
        if util.get_lab_mode() in ("enjoy", "eval"):
            return np.nan
        raise NotImplementedError

After Change

    @lab_api
    def train(self):
        """Implement algorithm train, or throw NotImplementedError"""
        if util.in_eval_lab_modes():
            return np.nan
        raise NotImplementedError

Instances