From b1ffaa296ec1789e102cb791d2097571fcaf1d5c Mon Sep 17 00:00:00 2001
From: Stanislav Boboň
Date: Fri, 12 Feb 2021 14:57:29 +0100
Subject: group vendors with more sanity

---
 sec_certs/dataset.py | 7 ++++---
 1 file 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):
-- 
cgit v1.3.1