cd50ec1c075fed30b19771879b818ba14ecfb816,test/test_saving.py,TestSaving,test_modules,#TestSaving#,33

Before Change

        self.start_tests(name="modules")

        // Remove directory if exists
        if os.path.exists(path=self.__class__.directory):
            for filename in os.listdir(path=self.__class__.directory):
                os.remove(path=os.path.join(self.__class__.directory, filename))
            os.rmdir(path=self.__class__.directory)

        agent, environment = self.prepare(config=dict(eager_mode=False))
        states = environment.reset()
        actions = agent.act(states=states)
        states, terminal, reward = environment.execute(actions=actions)

After Change

            files = set(os.listdir(path=directory))
            self.assertTrue(len(files), 2 * len(agent.model.this_submodules))
            for module in agent.model.this_submodules:
                self.assertTrue(module.name + ".index" in files)
                self.assertTrue(module.name + ".data-00000-of-00001" in files)

        agent.close()
        environment.close()

Instances