Minor corrections to format

Author: VisLab

hed/tools/analysis/tabular_summary.py

@@ -111,7 +111,7 @@ def get_summary(self, as_json=False) -> Union[dict, str]:
             "Value columns": value_cols,
             "Skip columns": self.skip_cols,
             "Files": self.files,
-            "Limit on categorical values": str(self.categorical_limit),
+            "Categorical limit": str(self.categorical_limit),
         }
         if as_json:
             return json.dumps(summary, indent=4)
@@ -180,7 +180,7 @@ def _update_categorical(self, tab_name, values, cat_counts):
         Parameters:
             tab_name (str): Name of a key indicating a categorical column.
             values (dict): A dictionary whose keys are unique categorical values.
-            cat_counts(list): A list with two elements: total count of values and number of entries.
+            cat_counts (list): A list with two elements: total count of values and number of entries.
 
         """
         if tab_name not in self.categorical_info:
@@ -311,6 +311,7 @@ def extract_summary(summary_info) -> "TabularSummary":
             value_cols=summary_info.get("Value columns", {}).keys(),
             skip_cols=summary_info.get("Skip columns", []),
             name=summary_info.get("Name", ""),
+            categorical_limit=summary_info.get("Categorical limit", None),
         )
         new_tab.value_info = summary_info.get("Value columns", {})
         new_tab.total_files = summary_info.get("Total files", 0)

tests/tools/analysis/test_tabular_summary.py

@@ -291,14 +291,14 @@ def test_categorical_limit_in_summary(self):
         dict1.update(stern_df)
 
         summary = dict1.get_summary(as_json=False)
-        self.assertIn("Limit on categorical values", summary)
-        self.assertEqual(summary["Limit on categorical values"], "10")
+        self.assertIn("Categorical limit", summary)
+        self.assertEqual(summary["Categorical limit"], "10")
 
         # Test with None
         dict2 = TabularSummary()
         dict2.update(stern_df)
         summary2 = dict2.get_summary(as_json=False)
-        self.assertEqual(summary2["Limit on categorical values"], "None")
+        self.assertEqual(summary2["Categorical limit"], "None")
 
     def test_categorical_limit_extract_summary(self):
         # Test that categorical_limit is preserved through extract_summary