# !pip install git+https://github.com/compute-tooling/compute-studio-kit.git@filespec
!conda install taxcalc -c pslmodels -y
import json
import fsspec
from cs_kit.filespec import CSFileSystem
with fsspec.open("cs://PSLmodels:Tax-Brain@47410/inputs/adjustment/policy/", "r") as f:
data = json.loads(f.read())
data
{'II_rt1': [{'year': 2019, 'value': 0.25}], 'II_rt2': [{'year': 2019, 'value': 0.25}], 'II_rt3': [{'year': 2019, 'value': 0.25}], 'II_rt4': [{'year': 2019, 'value': 0.25}], 'II_rt5': [{'year': 2019, 'value': 0.25}], 'II_rt6': [{'year': 2019, 'value': 0.25}], 'II_rt7': [{'year': 2019, 'value': 0.7}], 'II_brk6': [{'MARS': 'widow', 'year': 2019, 'value': 2000000}, {'MARS': 'headhh', 'year': 2019, 'value': 2000000}, {'MARS': 'mseparate', 'year': 2019, 'value': 2000000}, {'MARS': 'mjoint', 'year': 2019, 'value': 2000000}, {'MARS': 'single', 'year': 2019, 'value': 2000000}]}
import taxcalc
pol = taxcalc.Policy()
pol.adjust("cs://PSLmodels:Tax-Brain@47410/inputs/adjustment/policy")
OrderedDict([('II_rt1', [{'year': 2019, 'value': 0.25}]), ('II_rt2', [{'year': 2019, 'value': 0.25}]), ('II_rt3', [{'year': 2019, 'value': 0.25}]), ('II_rt4', [{'year': 2019, 'value': 0.25}]), ('II_rt5', [{'year': 2019, 'value': 0.25}]), ('II_rt6', [{'year': 2019, 'value': 0.25}]), ('II_brk6', [{'year': 2019, 'value': 2000000.0, 'MARS': 'widow'}, {'year': 2019, 'value': 2000000.0, 'MARS': 'headhh'}, {'year': 2019, 'value': 2000000.0, 'MARS': 'mseparate'}, {'year': 2019, 'value': 2000000.0, 'MARS': 'mjoint'}, {'year': 2019, 'value': 2000000.0, 'MARS': 'single'}]), ('II_rt7', [{'year': 2019, 'value': 0.7}])])
import io
import pandas as pd
import paramtools
data = paramtools.read_json("cs://PSLmodels:Tax-Brain@47410/outputs")
results = []
for output in data:
results.append(
pd.read_csv(io.StringIO(output["data"]))
)
results[0]
Unnamed: 0 | 2019 | 2020 | 2021 | 2022 | 2023 | 2024 | 2025 | 2026 | 2027 | 2028 | 2029 | |
---|---|---|---|---|---|---|---|---|---|---|---|---|
0 | Individual Income Tax Liability Change | $651.31 | $677.74 | $707.50 | $735.71 | $764.94 | $793.97 | $824.71 | $609.33 | $631.98 | $654.94 | $678.08 |
1 | Payroll Tax Liability Change | $0.00 | $0.00 | $0.00 | $0.00 | $0.00 | $0.00 | $0.00 | $0.00 | $0.00 | $0.00 | $0.00 |
2 | Combined Payroll and Individual Income Tax Lia... | $651.31 | $677.74 | $707.50 | $735.71 | $764.94 | $793.97 | $824.71 | $609.33 | $631.98 | $654.94 | $678.08 |
with fsspec.open("cs://PSLmodels:Tax-Brain@47410/owner") as f:
result = json.loads(f.read())
result
{'owner': 'MaxGhenis'}
with fsspec.open("cs://PSLmodels:Tax-Cruncher@520/title", api_token="your-api-token") as f:
result = json.loads(f.read())
result
{'title': 'Super secret sim'}