Skip to content
Snippets Groups Projects
Commit a67b7cb4 authored by ViaFerrata's avatar ViaFerrata
Browse files

- Added orcasong_contrib to docs

- Added command line tools to setup.py
parent f01cc29d
No related branches found
No related tags found
No related merge requests found
......@@ -57,7 +57,7 @@ autosummary_generate = True
# Document Python Code
autoapi_type = 'python'
autoapi_dirs = ['../orcasong']
autoapi_dirs = ['../orcasong', '../orcasong_contrib']
autoapi_options = [
'members', 'undoc-members'
# , 'private-members', 'special-members'
......
......@@ -344,9 +344,9 @@ def make_concatenate_and_shuffle_list_files(cfg):
+ ' ' + conc_outputfile_fpath)
def make_data_split():
def main():
"""
Main function.
Main function to make the data split.
"""
cfg = parse_input()
......@@ -379,4 +379,4 @@ def make_data_split():
if __name__ == '__main__':
make_data_split()
main()
......@@ -271,7 +271,7 @@ def shuffle_h5(filepath_input, tool=False, seed=42, delete=True, chunksize=None,
return output_file_shuffled
def shuffle_h5_tool():
def main():
"""
Frontend for the shuffle_h5 function that can be used in a bash environment.
......@@ -297,4 +297,4 @@ def shuffle_h5_tool():
if __name__ == '__main__':
shuffle_h5_tool()
main()
......@@ -15,7 +15,10 @@ setup(
install_requires=requirements,
packages=find_packages(),
include_package_data=True,
entry_points={'console_scripts': ['make_nn_images=orcasong.make_nn_images:main']},
entry_points={'console_scripts': ['make_nn_images=orcasong.make_nn_images:main',
'shuffle=orcasong_contrib.data_tools.shuffle.shuffle_h5:main',
'concatenate=orcasong_contrib.data_tools.concatenate.concatenate_h5:main',
'make_dsplit=orcasong_contrib.data_tools.make_data_split.make_data_split:main']},
setup_requires=['setuptools_scm'],
use_scm_version={
'write_to': 'orcasong/version.txt',
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment