diff options
Diffstat (limited to 'src/argaze/GazeAnalysis/NearestNeighborIndex.py')
-rw-r--r-- | src/argaze/GazeAnalysis/NearestNeighborIndex.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/argaze/GazeAnalysis/NearestNeighborIndex.py b/src/argaze/GazeAnalysis/NearestNeighborIndex.py index 1ca336f..1dc692e 100644 --- a/src/argaze/GazeAnalysis/NearestNeighborIndex.py +++ b/src/argaze/GazeAnalysis/NearestNeighborIndex.py @@ -11,7 +11,7 @@ __license__ = "BSD" from typing import TypeVar, Tuple, Any from dataclasses import dataclass, field -from argaze import GazeFeatures +from argaze import GazeFeatures, DataFeatures import numpy from scipy.spatial.distance import cdist @@ -35,6 +35,7 @@ class ScanPathAnalyzer(GazeFeatures.ScanPathAnalyzer): self.__nearest_neighbor_index = 0 + @DataFeatures.PipelineStep def analyze(self, scan_path: GazeFeatures.ScanPathType): assert(len(scan_path) > 1) |