Add Scanned Files to Results (#88)

Co-authored-by: Sean Morgan <[email protected]>
protectai · Jan 22, 2024 · 4f760ac · 4f760ac
1 parent 8dfa996
commit 4f760ac
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/modelscan/modelscan.py b/modelscan/modelscan.py
@@ -162,6 +162,11 @@ def _generate_results(self) -> Dict[str, Any]:
 
         report["errors"] = [str(error) for index, error in enumerate(self._errors)]
 
+        report["scanned"] = {"total_scanned": len(self._scanned)}
+        report["scanned"]["scanned_files"] = [
+            str(file_name) for file_name in self._scanned
+        ]
+
         report["skipped"] = {"total_skipped": len(self._skipped)}
         report["skipped"]["skipped_files"] = [
             str(file_name) for file_name in self._skipped