def _get_valid_kwargs(kwargs: Dict[str, Any]) -> Dict[str, Any]:
        """
        Get a valid kwargs to initialize a MILProtoExporter object.
        """
        return {
            "prog": kwargs["prog"],
            "weights_dir": kwargs["weights_dir"],
            "specification_version": kwargs["specification_version"],
        }

        Get a valid kwargs to initialize a MILProtoExporter object.
        
coremltools/converters/mil/backend/mil/load.py