def get_model()

in src/vw-serving/src/vw_serving/model_manager.py [0:0]

• 14 lines of code
• 5 McCabe index (conditional complexity)

    def get_model(self, disk_path=None, model_id=None):
        """
        Returns a tuple (str, str) of metadata string and model weights URL on disk
        """
        if disk_path:
            sagemaker_model_path = Path(disk_path)
            meta_files = list(sagemaker_model_path.rglob("vw.metadata"))
            if len(meta_files) == 0:
                raise CustomerError("'vw.metadata' not found in model files.")
            metadata_path = meta_files[0]

            model_files = list(sagemaker_model_path.rglob("vw.model"))
            if len(model_files) == 0:
                raise CustomerError("'vw.model' not found in model files.")
            model_path = model_files[0]
            return metadata_path.as_posix(), model_path.as_posix()
        elif model_id:
            return self._download_and_extract_model_tar_gz(model_id=model_id)