Fix Python 3 issue. See SlicerProstate#358.

tokjun · Oct 10, 2020 · 543b9ee · 543b9ee
1 parent e9da7e8
commit 543b9ee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/SliceTracker/SliceTrackerUtils/steps/plugins/charts.py b/SliceTracker/SliceTrackerUtils/steps/plugins/charts.py
@@ -36,7 +36,7 @@ def isTargetDisplacementChartDisplayable(self, selectedSeries):
     approvedResults = sorted([r for r in self.session.data.registrationResults.values() if r.approved],
                              key=lambda result: result.seriesNumber)
     nonSelectedApprovedResults = filter(lambda x: x.seriesNumber != selectedSeriesNumber, approvedResults)
-    if len(nonSelectedApprovedResults) == 0 or self.session.currentResult is None:
+    if len(list(nonSelectedApprovedResults)) == 0 or self.session.currentResult is None:
       return False
     return True
 
@@ -244,4 +244,4 @@ def createPlot(self, color, plotNumber):
     plotSeriesNode.SetMarkerSize(3 * plotSeriesNode.GetLineWidth())
 
     self._plotSeriesNodes.append(plotSeriesNode)
-    self._plotChartNode.AddAndObservePlotSeriesNodeID(plotSeriesNode.GetID())
+    self._plotChartNode.AddAndObservePlotSeriesNodeID(plotSeriesNode.GetID())
diff --git a/SliceTracker/SliceTrackerUtils/steps/zFrameRegistration.py b/SliceTracker/SliceTrackerUtils/steps/zFrameRegistration.py
@@ -101,7 +101,7 @@ def loadTemplateConfigFile(self):
 
     defaultTemplateFile = os.path.join(self.resourcesPath, "zframe", self.ZFRAME_TEMPLATE_CONFIG_FILE_NAME)
 
-    reader = csv.reader(open(defaultTemplateFile, 'rb'))
+    reader = csv.reader(open(defaultTemplateFile, newline=''))
     try:
       next(reader)
       for row in reader: