From dc0c6ed9f8b993e63f492f203d7d7080ab4c835c Mon Sep 17 00:00:00 2001 From: Richard Burton Date: Wed, 8 Apr 2020 16:39:05 +0100 Subject: Add PyArmNN to work with ArmNN API of 20.02 * Add Swig rules for generating python wrapper * Add documentation * Add tests and testing data Change-Id: If48eda08931514fa21e72214dfead2835f07237c Signed-off-by: Richard Burton Signed-off-by: Derek Lamberti --- python/pyarmnn/scripts/generate_docs.py | 50 +++++++++++++++++++++++++++++++++ 1 file changed, 50 insertions(+) create mode 100644 python/pyarmnn/scripts/generate_docs.py (limited to 'python/pyarmnn/scripts') diff --git a/python/pyarmnn/scripts/generate_docs.py b/python/pyarmnn/scripts/generate_docs.py new file mode 100644 index 0000000000..66eff6d52a --- /dev/null +++ b/python/pyarmnn/scripts/generate_docs.py @@ -0,0 +1,50 @@ +# Copyright © 2020 Arm Ltd. All rights reserved. +# SPDX-License-Identifier: MIT +"""Generate PyArmNN documentation.""" +import os +import tarfile + +import pyarmnn as ann +import shutil + +from typing import List, Union + +from pdoc.cli import main + + +def __copy_file_to_dir(file_paths: Union[List[str], str], target_dir_path: str): + file_paths = [] + file_paths + + if not (os.path.exists(target_dir_path) and os.path.isdir(target_dir_path)): + os.makedirs(target_dir_path) + + for file_path in file_paths: + if not (os.path.exists(file_path) and os.path.isfile(file_path)): + raise RuntimeError('Not a file: {}'.format(file_path)) + + file_name = os.path.basename(file_path) + shutil.copyfile(file_path, os.path.join(str(target_dir_path), file_name)) + + +def copy_doc_images(): + __copy_file_to_dir(file_paths=['./images/pyarmnn.png'], + target_dir_path='docs/pyarmnn/images') + + +def archive_docs(path, version): + + output_filename = f'pyarmnn_docs-{version}.tar' + + with tarfile.open(output_filename, "w") as tar: + tar.add(path) + + +if __name__ == "__main__": + with open('./README.md', 'r') as readme_file: + top_level_pyarmnn_doc = ''.join(readme_file.readlines()) + ann.__doc__ = top_level_pyarmnn_doc + + main() + + copy_doc_images() + archive_docs('./docs', ann.__version__) -- cgit v1.2.1