dataref export: include environment variables

1 files changed, 3 insertions, 0 deletions

diff --git a/lib/cli.py b/lib/cli.py
index ed25d62..ff58099 100644
--- a/lib/cli.py
+++ b/lib/cli.py
@@ -336,6 +336,9 @@ def model_quality_table(
 
 def export_dataref(dref_file, dref, precision=None):
     with open(dref_file, "w") as f:
+        for k, v in sorted(os.environ.items(), key=lambda kv: kv[1]):
+            if k.startswith("DFATOOL_"):
+                print(f"% {k}='{v}'", file=f)
         for arg in sys.argv:
             print(f"% {arg}", file=f)
         for k, v in sorted(dref.items()):