diff --git a/src/comsar/_tracks/timbre.py b/src/comsar/_tracks/timbre.py
index a0041fcae126d63f2e3617fe6bcc68a567244427..8a91ef918b04be13790fbc664478029950513af1 100644
--- a/src/comsar/_tracks/timbre.py
+++ b/src/comsar/_tracks/timbre.py
@@ -16,7 +16,7 @@ from apollon.tools import time_stamp
import comsar
from comsar._tracks.utilities import (TrackMeta, TrackResult,
- TimbreTrackParams)
+ TimbreTrackParams, TimbreTrackAltParams)
STFT_DEFAULT = container.StftParams(fps=44100, window='hamming', n_fft=None,
@@ -125,9 +125,8 @@ class TimbreTrackAlt:
"""
Args:
"""
- self.params = TimbreTrackParams(stft_params or STFT_DEFAULT,
- corr_dim_params or CORR_DIM_DEFAULT,
- corr_gram_params or CORR_GRAM_DEFAULT)
+ self.params = TimbreTrackAltParams(stft_params or STFT_DEFAULT,
+ corr_dim_params or CORR_DIM_DEFAULT)
self.cutter = Segmentation(self.params.stft.n_perseg,
self.params.stft.n_overlap,
diff --git a/src/comsar/_tracks/utilities.py b/src/comsar/_tracks/utilities.py
index 549acbfe4f438eb3564b23c6c7ba984be62ef864..5b54ee26160585c3e5930ab5b3a26f22c38ae818 100644
--- a/src/comsar/_tracks/utilities.py
+++ b/src/comsar/_tracks/utilities.py
@@ -42,6 +42,13 @@ class TimbreTrackParams(TrackParams):
corr_gram: signal.container.CorrGramParams
+@dataclass
+class TimbreTrackAltParams(TrackParams):
+ """Parameter set for TimbreTrack"""
+ stft: signal.container.StftParams
+ corr_dim: signal.container.CorrDimParams
+
+
class TrackResult:
"""Provide track results."""
def __init__(self, meta: TrackMeta, params: TrackParams,
diff --git a/src/comsar/tracks.py b/src/comsar/tracks.py
index 969dbf1857d1137ec6514bb1dac2cb821c025f0a..03c48abb0fa2036324a75146137bf1a213a46f3b 100644
--- a/src/comsar/tracks.py
+++ b/src/comsar/tracks.py
@@ -1 +1,2 @@
from . _tracks.timbre import TimbreTrack
+from . _tracks.timbre import TimbreTrackAlt