add version to settings

protectai · Jan 19, 2024 · 843ee52 · 843ee52
1 parent 0ed0818
commit 843ee52
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/modelscan/settings.py b/modelscan/settings.py
@@ -2,12 +2,15 @@
 
 from typing import Any
 
+from modelscan._version import __version__
+
 DEFAULT_REPORTING_MODULES = {
     "console": "modelscan.reports.ConsoleReport",
     "json": "modelscan.reports.JSONReport",
 }
 
 DEFAULT_SETTINGS = {
+    "modelscan_version": __version__,
     "supported_zip_extensions": [".zip", ".npz"],
     "scanners": {
         "modelscan.scanners.H5LambdaDetectScan": {