format

Signed-off-by: Trevor Madge <tmadge@paloaltonetworks.com>

Author: madgetr

modelscan/scanners/saved_model/scan.py

@@ -57,8 +57,10 @@ def _load_safe_operators() -> List[str]:
         try:
             with open(operators_file, "r") as f:
                 data = json.load(f)
-                return  [
-                    operator for operator in list(data.get("operators", [])) if operator[0] != "_"
+                return [
+                    operator
+                    for operator in list(data.get("operators", []))
+                    if operator[0] != "_"
                 ]
 
         except (FileNotFoundError, json.JSONDecodeError) as e:

scripts/update_tensorflow_operators.py

@@ -20,9 +20,7 @@
 from typing import List, Dict, Any
 
 
-def update_operators_json(
-    json_path: Path
-) -> None:
+def update_operators_json(json_path: Path) -> None:
     """
     Update the JSON file with new version and operators.
 
@@ -41,14 +39,17 @@ def update_operators_json(
     for op in tf.raw_ops.__dict__.keys():
         operators.append(op)
     data = {}
-    data['description'] = "List of known TensorFlow raw operators from tensorflow.raw_ops.__dict__.keys()"
-    data['version'] = version
-    data['operators'] = operators
+    data["description"] = (
+        "List of known TensorFlow raw operators from tensorflow.raw_ops.__dict__.keys()"
+    )
+    data["version"] = version
+    data["operators"] = operators
 
     # Write updated file with nice formatting
-    with open(json_path, 'w') as f:
+    with open(json_path, "w") as f:
         json.dump(data, f, indent=2)
 
+
 def main():
     """Main entry point."""
     # Determine paths