1 files changed, 10 insertions, 7 deletions
diff --git a/examples/colormapDialog.py b/examples/colormapDialog.py
index c9e7c35..8eeb657 100644
--- a/examples/colormapDialog.py
+++ b/examples/colormapDialog.py
@@ -1,7 +1,6 @@
-# coding: utf-8
 # /*##########################################################################
 #
-# Copyright (c) 2017-2018 European Synchrotron Radiation Facility
+# Copyright (c) 2017-2021 European Synchrotron Radiation Facility
 #
 # Permission is hereby granted, free of charge, to any person obtaining a copy
 # of this software and associated documentation files (the "Software"), to deal
@@ -140,7 +139,7 @@ class ColormapDialogExample(qt.QMainWindow):
         button = qt.QPushButton("Negative to positive")
         button.clicked.connect(self.setDataFromNegToPos)
         layout.addWidget(button)
-        button = qt.QPushButton("Only non finite values")
+        button = qt.QPushButton("With non finite values")
         button.clicked.connect(self.setDataWithNonFinite)
         layout.addWidget(button)
 
@@ -235,10 +234,12 @@ class ColormapDialogExample(qt.QMainWindow):
 
     def setSheppLoganPhantom(self):
         from silx.image import phantomgenerator
+
         data = phantomgenerator.PhantomGenerator.get2DPhantomSheppLogan(256)
         data = data * 1000
         if scipy is not None:
             from scipy import ndimage
+
             data = ndimage.gaussian_filter(data, sigma=20)
         data = numpy.random.poisson(data)
         self.data = data
@@ -246,7 +247,7 @@ class ColormapDialogExample(qt.QMainWindow):
             dialog.setData(self.data)
 
     def setDataFromNegToPos(self):
-        data = numpy.ones((50,50))
+        data = numpy.ones((50, 50))
         data = numpy.random.poisson(data)
         self.data = data - 0.5
         for dialog in self.colorDialogs:
@@ -254,12 +255,14 @@ class ColormapDialogExample(qt.QMainWindow):
 
     def setDataWithNonFinite(self):
         from silx.image import phantomgenerator
+
         data = phantomgenerator.PhantomGenerator.get2DPhantomSheppLogan(256)
         data = data * 1000
         if scipy is not None:
             from scipy import ndimage
+
             data = ndimage.gaussian_filter(data, sigma=20)
-        data = numpy.random.poisson(data)
+        data = numpy.random.poisson(data).astype(numpy.float32)
         data[10] = float("nan")
         data[50] = float("+inf")
         data[100] = float("-inf")
@@ -275,8 +278,8 @@ def main():
     example = ColormapDialogExample()
     example.show()
 
-    app.exec_()
+    app.exec()
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()