group vendors with more sanity

1 files changed, 4 insertions, 3 deletions

diff --git a/sec_certs/dataset.py b/sec_certs/dataset.py
index 9c720a71..7b8a0f73 100644
--- a/sec_certs/dataset.py
+++ b/sec_certs/dataset.py
@@ -864,13 +864,14 @@ class FIPSDataset(Dataset, ComplexSerializableType):
         dset.root_dir = input_path.parent.absolute()
         return dset
 
-    def group_vendors(self):
+    def group_vendors(self) -> Dict:
         vendors = {}
-        v = set([x.vendor.lower() for x in self.certs.values()])
+        v = {x.vendor.lower() for x in self.certs.values()}
         v = sorted(v, key=FIPSCertificate.get_compare)
         for prefix, a in groupby(v, key=FIPSCertificate.get_compare):
             vendors[prefix] = list(a)
-        print(json.dumps(vendors, indent=4))
+
+        return vendors
 
 
 class FIPSAlgorithmDataset(Dataset, ComplexSerializableType):