addbe43e93bea82a3eda2b8051df6b7e40fd9ec1,studio/model.py,Experiment,get_model,#Experiment#,61

Before Change

        if self.info.get("type") == "keras":
            hdf5_files = [
                (p, os.path.getmtime(p))
                for p in glob.glob(os.path.join(self.model_dir, "*.hdf*"))]

            last_checkpoint = max(hdf5_files, key=lambda t: t[1])[0]
            return KerasModelWrapper(last_checkpoint)

After Change

                (p, os.path.getmtime(p))
                for p in
                glob.glob(self.artifacts["modeldir"] + "/*.hdf*") +
                glob.glob(self.artifacts["modeldir"] + "/*.h5")]

            last_checkpoint = max(hdf5_files, key=lambda t: t[1])[0]
            return KerasModelWrapper(last_checkpoint)

Instances