feat(cli): add extra dataset stats

reinfer · Sep 3, 2024 · 4304193 · 4304193
1 parent fa87853
commit 4304193
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/api/src/resources/dataset.rs b/api/src/resources/dataset.rs
@@ -55,6 +55,9 @@ pub struct Dataset {
 pub struct DatasetStats {
     pub total_verbatims: NotNan<f64>,
     pub validation: Option<ValidationResponse>,
+    pub number_of_labels: usize,
+    pub number_of_fields: usize,
+    pub number_of_extraction_defs: usize,
 }
 
 #[derive(Debug, Clone, Deserialize, Serialize, PartialEq)]

diff --git a/cli/src/commands/get/datasets.rs b/cli/src/commands/get/datasets.rs
@@ -80,6 +80,13 @@ pub fn get(
                         stats: DatasetStats {
                             total_verbatims: unfiltered_stats.num_comments,
                             validation: validation_response.ok(),
+                            number_of_labels: dataset.label_defs.len(),
+                            number_of_fields: dataset.entity_defs.len(),
+                            number_of_extraction_defs: dataset
+                                .label_defs
+                                .iter()
+                                .map(|l| if l.moon_form.is_some() { 1 } else { 0 })
+                                .sum(),
                         },
                     })
                 };