add scanned files to results report

protectai · Jan 19, 2024 · 4e3e6d6 · 4e3e6d6
1 parent 3ea3473
commit 4e3e6d6
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