assignment only inference

seung-lab · May 24, 2024 · 541a5ae · 541a5ae
1 parent 7fde69b
commit 541a5ae
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/dags/synaptor_dags.py b/dags/synaptor_dags.py
@@ -285,6 +285,12 @@ def add_task(
     CPUTask("remap"),
 ]
 
+file_assignmentonly = [
+    GPUTask("chunk_edges"),
+    CPUTask("merge_edges"),
+    CPUTask("remap")
+]
+
 # =========================================
 # DAG definition
 dag = DAG(
@@ -297,6 +303,7 @@ def add_task(
 workflows = {
     "Segmentation": {"File": file_segmentation, "Database": db_segmentation},
     "Segmentation+Assignment": {"File": file_assignment, "Database": db_assignment},
+    "Assignment": {"File": file_assignmentonly, "Database": file_assignmentonly}
 }
 
 fill_dag(