Skip to content

Commit

Permalink
adapt hd-bet entrypoint to be compatible with pyproject.toml
Browse files Browse the repository at this point in the history
This will make it possible to switch from setup.py to pyproject.toml
in a shortly upcoming commit
  • Loading branch information
ebrahimebrahim committed May 29, 2024
1 parent f18163c commit d71686a
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion HD_BET/hd-bet → HD_BET/hd_bet_cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
import HD_BET


if __name__ == "__main__":
def main():
print("\n########################")
print("If you are using hd-bet, please cite the following paper:")
print("Isensee F, Schell M, Tursunova I, Brugnara G, Bonekamp D, Neuberger U, Wick A, Schlemmer HP, Heiland S, Wick W,"
Expand Down Expand Up @@ -132,3 +132,6 @@
raise ValueError("Unknown value for bet: %s. Expected: 0 or 1" % str(pp))

run_hd_bet(input_files, output_files, mode, config_file, device, pp, tta, save_mask, overwrite_existing, bet)

if __name__ == "__main__":
main()

0 comments on commit d71686a

Please sign in to comment.