From c8849301fe671a32441ce49ec29ef5634fbc292c Mon Sep 17 00:00:00 2001 From: frankknoll Date: Fri, 17 Nov 2023 16:44:59 +0100 Subject: [PATCH] refactoring --- .../MultiLineFitting/MultiLineFitter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/SymptomsCausedByVaccines/MultiLineFitting/MultiLineFitter.py b/src/SymptomsCausedByVaccines/MultiLineFitting/MultiLineFitter.py index 2b7699bf501..a370fc4672b 100644 --- a/src/SymptomsCausedByVaccines/MultiLineFitting/MultiLineFitter.py +++ b/src/SymptomsCausedByVaccines/MultiLineFitting/MultiLineFitter.py @@ -14,7 +14,7 @@ class MultiLineFitter: preferenceMatrix = MultiLineFitter._createPreferenceMatrix(points, lines, consensusThreshold) _, preferenceMatrix4Clusters = MultiLineFitter._createClusters(preferenceMatrix) lineIndexes = MultiLineFitter._getLineIndexes(preferenceMatrix4Clusters) - return [lines[lineIndex] for lineIndex in lineIndexes] + return np.array(lines)[lineIndexes] @staticmethod def _createPreferenceMatrix(points, lines, consensusThreshold):