From 3f983ae5f7d5bf3cd60c7d694bfa09089e09ff15 Mon Sep 17 00:00:00 2001 From: wasserth Date: Wed, 30 Aug 2023 10:33:55 +0200 Subject: [PATCH] bugfix missing import --- setup.py | 7 ++++++- tests_win.py | 4 +++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 32fe1ac81..c4228ac23 100644 --- a/setup.py +++ b/setup.py @@ -53,5 +53,10 @@ 'bin/TotalSegmentator', 'bin/totalseg_combine_masks', 'bin/crop_to_body', 'bin/totalseg_import_weights', 'bin/totalseg_download_weights', 'bin/totalseg_setup_manually', 'bin/totalseg_set_license' - ] + ], + # entry_points={ + # 'console_scripts': [ + # 'TotalSegmentator=totalsegmentator:main' + # ], + # }, ) diff --git a/tests_win.py b/tests_win.py index f32f9c37b..0bee593cd 100755 --- a/tests_win.py +++ b/tests_win.py @@ -5,6 +5,8 @@ import shutil import subprocess +from totalsegmentator.python_api import totalsegmentator + if __name__ == "__main__": @@ -16,7 +18,7 @@ # Test terminal # Test organ predictions - fast - multilabel # makes correct path for windows and linux. Only required for terminal call. Within python - # I can always / and i will correctly be interpreted on windows + # I can always use / and it will correctly be interpreted on windows file_in = os.path.join("tests", "reference_files", "example_ct_sm.nii.gz") print("File_in path:") print(file_in)