bugfix missing import

wasserth · Aug 30, 2023 · 3f983ae · 3f983ae
1 parent 0451f37
commit 3f983ae
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git 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
@@ -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)