Lower employment income by 10%

.github/review_pull_request.py

@@ -1,6 +1,6 @@
 import pandas as pd
 from policyengine_us_data.data_storage import STORAGE_FOLDER
-from IPython.display import Markdown
+from policyengine_us_data.utils.github import set_pr_auto_review_comment
 
 def main():
     df = pd.read_csv(STORAGE_FOLDER / "evaluation.csv")
@@ -11,11 +11,15 @@ def main():
     diff = (most_recent_rows - second_most_recent_rows)
     # Convert to df
     diff = diff.reset_index()
+    diff.Total = diff.Total.apply(lambda x: f"{x:+.1f}")
+    diff = diff[diff.Variable == "household_net_income"].set_index("Time period")[["Total"]].T
     table = diff.to_markdown(index=False)
 
-    review_text = f"""## National projection changes\n\n{table}"""
+    review_text = f"""## National projection changes\n\nThis pull request makes the following changes to economic estimates.\n\n### Household net income\n\n{table}"""
 
     print(review_text)
 
+    set_pr_auto_review_comment(review_text)
+
 if __name__ == "__main__":
     main()

.github/upload_evaluation.py

@@ -3,7 +3,6 @@
 from policyengine_us_data.data_storage import STORAGE_FOLDER
 
 if __name__ == "__main__":
-    main()
     upload(
         "policyengine",
         "policyengine-us-data",

.github/workflows/pull_request.yaml

@@ -11,6 +11,7 @@ jobs:
     runs-on: ubuntu-latest
     env:
       GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+      GITHUB_PR_NUMBER: ${{ github.event.number }}
 
     steps:
       - name: Checkout code

policyengine_us_data/datasets/cps/policyengine_cps.py

@@ -204,7 +204,7 @@ def add_personal_income_variables(
     assert isinstance(p, dict)
 
     # Assign CPS variables.
-    cps["employment_income"] = person.WSAL_VAL
+    cps["employment_income"] = person.WSAL_VAL * 0.9
 
     cps["weekly_hours_worked"] = person.HRSWK * person.WKSWORK / 52
 

policyengine_us_data/utils/github.py

@@ -79,3 +79,24 @@ def upload(
         )
 
     return response.json()
+
+def set_pr_auto_review_comment(text: str):
+    # On a pull request, set a review comment with the given text.
+
+    pr_number = os.environ["GITHUB_PR_NUMBER"]
+
+    url = f"https://api.github.com/repos/{os.environ['GITHUB_REPOSITORY']}/pulls/{pr_number}/reviews"
+
+    response = requests.post(
+        url,
+        headers=auth_headers,
+        json={
+            "body": text,
+            "event": "COMMENT",
+        },
+    )
+
+    if response.status_code != 200:
+        raise ValueError(
+            f"Invalid response code {response.status_code} for url {url}. Received: {response.text}"
+        )

setup.py

@@ -19,6 +19,7 @@
             "tqdm",
             "requests",
             "policyengine_us",
+            "tabulate",
         ],
     },
 )