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 ...@@ -57,7 +57,7 @@ autosummary_generate = True
# Document Python Code # Document Python Code
autoapi_type = 'python' autoapi_type = 'python'
autoapi_dirs = ['../orcasong'] autoapi_dirs = ['../orcasong', '../orcasong_contrib']
autoapi_options = [ autoapi_options = [
'members', 'undoc-members' 'members', 'undoc-members'
# , 'private-members', 'special-members' # , 'private-members', 'special-members'
......
...@@ -344,9 +344,9 @@ def make_concatenate_and_shuffle_list_files(cfg): ...@@ -344,9 +344,9 @@ def make_concatenate_and_shuffle_list_files(cfg):
+ ' ' + conc_outputfile_fpath) + ' ' + conc_outputfile_fpath)
def make_data_split(): def main():
""" """
Main function. Main function to make the data split.
""" """
cfg = parse_input() cfg = parse_input()
...@@ -379,4 +379,4 @@ def make_data_split(): ...@@ -379,4 +379,4 @@ def make_data_split():
if __name__ == '__main__': if __name__ == '__main__':
make_data_split() main()
...@@ -271,7 +271,7 @@ def shuffle_h5(filepath_input, tool=False, seed=42, delete=True, chunksize=None, ...@@ -271,7 +271,7 @@ def shuffle_h5(filepath_input, tool=False, seed=42, delete=True, chunksize=None,
return output_file_shuffled return output_file_shuffled
def shuffle_h5_tool(): def main():
""" """
Frontend for the shuffle_h5 function that can be used in a bash environment. Frontend for the shuffle_h5 function that can be used in a bash environment.
...@@ -297,4 +297,4 @@ def shuffle_h5_tool(): ...@@ -297,4 +297,4 @@ def shuffle_h5_tool():
if __name__ == '__main__': if __name__ == '__main__':
shuffle_h5_tool() main()
...@@ -15,7 +15,10 @@ setup( ...@@ -15,7 +15,10 @@ setup(
install_requires=requirements, install_requires=requirements,
packages=find_packages(), packages=find_packages(),
include_package_data=True, 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'], setup_requires=['setuptools_scm'],
use_scm_version={ use_scm_version={
'write_to': 'orcasong/version.txt', '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