26453d58671e01410b22287a723f76d6bb98803f,optuna/storages/in_memory.py,InMemoryStorage,set_trial_system_attr,#InMemoryStorage#Any#Any#Any#,325

Before Change

        // type: (int, str, Any) -> None

        with self._lock:
            self.check_trial_is_updatable(trial_id, self.trials[trial_id].state)

            trial = copy.copy(self.trials[trial_id])
            trial.system_attrs = copy.copy(trial.system_attrs)
            trial.system_attrs[key] = value
            self.trials[trial_id] = trial

    def get_trial(self, trial_id):
        // type: (int) -> FrozenTrial

After Change

            trial.user_attrs[key] = value
            self._set_trial(trial_id, trial)

    def set_trial_system_attr(self, trial_id, key, value):
        // type: (int, str, Any) -> None

        trial = self.get_trial(trial_id)
        self.check_trial_is_updatable(trial_id, trial.state)

        with self._lock:
            self.check_trial_is_updatable(trial_id, trial.state)

            trial = copy.copy(trial)
            trial.system_attrs = copy.copy(trial.system_attrs)
            trial.system_attrs[key] = value
            self._set_trial(trial_id, trial)

Instances