4bc348f58129abfedd9d27aaf2d301fb75fc5e75,optuna/storages/in_memory.py,InMemoryStorage,set_trial_system_attr,#InMemoryStorage#Any#Any#Any#,380

Before Change

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

        self._check_trial_id(trial_id)
        self.check_trial_is_updatable(trial_id, self._trials[trial_id].state)

        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

    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._trials[trial_id] = trial

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

Instances