Skip to content

Commit 56880f2

Browse files
fix accessing rarity value in lightness
1 parent f57d665 commit 56880f2

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

pyeyesweb/analysis_primitives/rarity.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ class Rarity:
77

88
def __call__(self, sliding_window: SlidingWindow, alpha: float = 0.5) -> dict:
99
if not sliding_window.is_full():
10-
return np.nan
10+
return {"rarity": np.nan}
1111

1212
samples, _ = sliding_window.to_array()
1313
n_samples = len(samples)

pyeyesweb/mid_level/lightness.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,9 @@ def __call__(self, velocity, alpha: float = 0.5) -> dict:
3535

3636
self.sliding_window.append([1.0 - weight_index]) # ora sliding_window contiene floats
3737

38-
lightness = self.rarity(self.sliding_window, alpha=alpha) # Rarity riceve una sequenza di numeri
38+
lightness = self.rarity(self.sliding_window, alpha=alpha)
3939

40-
return {"lightness": lightness}
40+
return {"lightness": lightness.get("rarity", 0)}
4141

4242
def _filter_signal(self, signal):
4343
if not self.use_filter:

0 commit comments

Comments
 (0)