Merge pull request #7486 from ggbecker/fix-up-fix_rules-script

Fix up for the fix_rules.py script in add-cce command
ComplianceAsCode · Aug 30, 2021 · 59b0c27 · 59b0c27
2 parents 9a01c85 + 93b8241
commit 59b0c27
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ssg/cce.py b/ssg/cce.py
@@ -1,4 +1,6 @@
 import re
+import random
+import os
 
 
 CCE_POOLS = dict()

diff --git a/utils/fix_rules.py b/utils/fix_rules.py
@@ -525,7 +525,7 @@ def fix_file_prompt(path, product_yaml, func, args):
 
 def add_cce(args, product_yaml):
     directory = os.path.join(args.root, args.subdirectory)
-    cce_pool = cce.CCE_POOLS[args.cce_pool]
+    cce_pool = cce.CCE_POOLS[args.cce_pool]()
     return _add_cce(directory, cce_pool, args.rule, product_yaml, args)