瀏覽代碼

updated segmentation query entry form to match nnUNET

Andrej Studen 1 年之前
父節點
當前提交
05b5533727
共有 1 個文件被更改,包括 8 次插入3 次删除
  1. 8 3
      pythonScripts/runSegmentationDM.py

+ 8 - 3
pythonScripts/runSegmentationDM.py

@@ -257,7 +257,10 @@ def doSegmentation(parameterFile):
 
         #update database
         copyFields=[pars['participantField'],'SequenceNum','patientCode','visitCode']
+        row['SequenceNum']+=0.001*float(pars['versionNumber'])
         filters=[{'variable':v,'value':str(row[v]),'oper':'eq'} for v in copyFields]
+        filters.append({'variable':'Version','value':pars['version'],'oper':'eq'})
+
         ds1=db.selectRows(pars['project'],pars['segmentationSchema'],pars['segmentationQuery'],filters)
         if len(ds1['rows'])>0:
             mode='update'
@@ -265,9 +268,11 @@ def doSegmentation(parameterFile):
         else:
             mode='insert'
             outRow={v:row[v] for v in copyFields}
-        outRow[pars['version']]= outName
-        db.modifyRows(mode,pars['project'],pars['segmentationSchema'],pars['segmentationQuery'],[outRow])
-        #pull results back to LabKey
+
+        outRow['Version']= pars['version']
+        outRow['Segmentation']= outName
+        print(db.modifyRows(mode,pars['project'],pars['segmentationSchema'],pars['segmentationQuery'],[outRow]))
+        #push results back to LabKey
         i+=1
         if i==1:
             break