Merge branch 'main' into spine_cropper

StanfordMIMI · Oct 13, 2023 · 1189af9 · 1189af9
2 parents 7c43269 + e8b6e9f
commit 1189af9
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 8 deletions.
diff --git a/bin/C2C-slurm b/bin/C2C-slurm
@@ -13,7 +13,7 @@ def submit_command(command):
     subprocess.run(command.split(" "), check=True, capture_output=False)
 
 
-def python_submit(command, node="amalfi"):
+def python_submit(command, node="siena"):
     bash_file = open("./slurm.sh", "w")
     bash_file.write(f"#!/bin/bash\n{command}")
     bash_file.close()

diff --git a/comp2comp/spine/spine.py b/comp2comp/spine/spine.py
@@ -16,12 +16,6 @@
 import wget
 from PIL import Image
 
-# from totalsegmentator.libs import (
-#     download_pretrained_weights,
-#     nostdout,
-#     setup_nnunet,
-# )
-
 from totalsegmentatorv2.python_api import totalsegmentator
 
 from comp2comp.inference_class_base import InferenceClass
@@ -30,6 +24,14 @@
 from comp2comp.visualization.dicom import to_dicom
 from comp2comp.io import io_utils
 
+# from totalsegmentator.libs import (
+#     download_pretrained_weights,
+#     nostdout,
+#     setup_nnunet,
+# )
+
+
+
 
 class SpineSegmentation(InferenceClass):
     """Spine segmentation."""

diff --git a/comp2comp/utils/process.py b/comp2comp/utils/process.py
@@ -5,10 +5,11 @@
 import os
 import shutil
 import sys
+import time
 import traceback
 from datetime import datetime
 from pathlib import Path
-import time
+
 
 from comp2comp.io import io_utils