13986978d4545aa429a7fc233d8e39718d52e255,studio/keyvalue_provider.py,KeyValueProvider,checkpoint_experiment,#KeyValueProvider#Any#Any#,223

Before Change

        experiment_dict = self._get(self._get_experiments_keybase() + key)

        checkpoint_threads = [
            Thread(
                target=self.store.put_artifact,
                args=(art,))
            for _, art in six.iteritems(experiment.artifacts)
            if art["mutable"] and art.get("local")]

        for t in checkpoint_threads:
            t.start()

        checkpoint_time = time.time()
        experiment.time_last_checkpoint = checkpoint_time
        experiment_dict["time_last_checkpoint"] = checkpoint_time

After Change

        with ThreadPoolExecutor(max_workers=8) as executor:
            for _, art in six.iteritems(experiment.artifacts):
                if art["mutable"] and art.get("local"):
                    workers.append(executor.submit(self.store.put_artifact, art))
            wait(workers)

        for worker in workers:

Instances