f9540d6da262c88cd4dfa2c3d9e8c2e570ea2f11,mushroom_rl/algorithms/agent.py,Agent,save,#Agent#Any#,130

Before Change

        )
        self._save_pickle(PurePath(path, "agent.config"), agent_config)

        for att, method in self._save_attributes.items():
            attribute = getattr(self, att) if hasattr(self, att) else None
            save_method = getattr(self, "_save_{}".format(method)) if hasattr(
                self, "_save_{}".format(method)) else None
            if attribute is None:
                continue
            elif save_method is None:
                raise NotImplementedError(
                    "Method _save_{} is not implemented for class "{}"".format(
                        method, self.__class__.__name__)
                )
            else:
                save_method(PurePath(path, "{}.{}".format(att, method)),
                            attribute)

    def copy(self):
        
        Returns:
             A deepcopy of the agent.

After Change

        path = Path(path)
        path.parent.mkdir(parents=True, exist_ok=True)

        with ZipFile(path, "w") as zip_file:
            agent_config = dict(
                type=type(self),
                save_attributes=self._save_attributes
            )

            self._save_pickle(zip_file, "agent.config", agent_config)

            for att, method in self._save_attributes.items():
                attribute = getattr(self, att) if hasattr(self, att) else None
                save_method = getattr(self, "_save_{}".format(method)) if hasattr(
                    self, "_save_{}".format(method)) else None
                if attribute is None:
                    continue
                elif save_method is None:
                    raise NotImplementedError(
                        "Method _save_{} is not implemented for class "{}"".format(
                            method, self.__class__.__name__)
                    )
                else:
                    file_name = "{}.{}".format(att, method)
                    save_method(zip_file, file_name, attribute)

    def copy(self):
        
        Returns:
             A deepcopy of the agent.

Instances