Merge branch 'comparison_board' of https://github.com/SainsburyWellco…

…meCentre/NeuralPlayground into comparison_board
SainsburyWellcomeCentre · Jul 25, 2023 · 519a261 · 519a261
2 parents 83d9014 + 5b26144
commit 519a261
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neuralplayground/comparison/metrics.py b/neuralplayground/comparison/metrics.py
@@ -528,7 +528,7 @@ def gridness(self, image, step):
             # get the correlation between the original and rotated images
             rotationalCorrVals[angle] = stats.pearsonr(autoCorrMiddleRescaled.ravel()[~allNans], rotatedA.ravel()[~allNans])[0]
             rotationArr[idx] = rotationalCorrVals[angle]
-        gridscore = np.min((rotationalCorrVals[60], rotationalCorrVals[120])) - np.max(
+        gridscore = np.nanmin((rotationalCorrVals[60], rotationalCorrVals[120])) - np.nanmax(
             (rotationalCorrVals[150], rotationalCorrVals[30], rotationalCorrVals[90])
         )
         return gridscore, rotationalCorrVals, rotationArr