mf

src/autometadata/__init__.py

@@ -89,6 +89,8 @@ def persist_ir_metadata(output_directory: Path, codecarbon_tracker: Optional[Emi
     collected_meta_data["git"] = collect_git_repo_metadata()
     collected_meta_data["cpuinfo"] = get_cpu_info()
     collected_meta_data["gpus"] = get_gpu_info()
+    executed_file = _executed_file_from_stacktrace()
+    collected_meta_data["file"] = {"name": executed_file.name, "content": open(executed_file, 'r').read()}
     if codecarbon_tracker:
         collected_meta_data["codecarbon_emissions"] = json.loads(codecarbon_tracker.final_emissions_data.toJSON())
 

test/PythonScriptApprovalTests.test_for_valid_git_repo.approved.json

@@ -2,6 +2,10 @@
     "cpuinfo": {
         "arch": "X86_64"
     },
+    "file": {
+        "content": "#!/usr/bin/env python3\nfrom pathlib import Path\n\nimport click\nimport pandas as pd\nimport pyterrier as pt\n\nfrom autometadata import persist_ir_metadata\n\n\n@click.command()\n@click.argument(\"output-directory\", type=Path)\ndef main(output_directory):\n    run = pd.DataFrame(\n        [\n            {\"qid\": \"q-1\", \"docno\": \"doc-01\", \"rank\": 1, \"score\": 10},\n            {\"qid\": \"q-1\", \"docno\": \"doc-02\", \"rank\": 2, \"score\": 9},\n            {\"qid\": \"q-1\", \"docno\": \"doc-03\", \"rank\": 3, \"score\": 8},\n        ]\n    )\n    output_directory.mkdir(exist_ok=True, parents=True)\n    pt.io.write_results(run, output_directory / \"run.txt\", format=\"trec\")\n    persist_ir_metadata(output_directory)\n\n\nif __name__ == \"__main__\":\n    main()\n",
+        "name": "example-script.py"
+    },
     "git": {
         "active_branch": "main",
         "commit": "a810f6948292c8c6ffef4fc0c8172887a8715c8d",