Merge pull request #7 from SpikeInterface/qm-names

qm names list
SpikeInterface · Jan 15, 2024 · a261169 · a261169
2 parents bea6adc + bf57985
commit a261169
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/spikeinterface_pipelines/postprocessing/params.py b/src/spikeinterface_pipelines/postprocessing/params.py
@@ -94,10 +94,12 @@ class QMParams(BaseModel):
     nn_noise_overlap: NNIsolation = Field(default=NNIsolation(), description="Nearest neighbor noise overlap.")
 
 
-# TODO - fill in metric_names defauklt value with all the metric names
 class QualityMetrics(BaseModel):
     qm_params: QMParams = Field(default=QMParams(), description="Quality metric parameters.")
-    metric_names: Optional[List[str]] = Field(default=None, description="List of metric names to compute. If None, all available metrics are computed.")
+    metric_names: List[str] = Field(
+        default=["presence_ratio", "snr", "isi_violation", "rp_violation", "sliding_rp_violation", "amplitude_cutoff", "amplitude_median", "nearest_neighbor", "nn_isolation", "nn_noise_overlap"],
+        description="List of metric names to compute. If None, all available metrics are computed."
+    )
     n_jobs: int = Field(default=1, description="Number of jobs.")