owlplanner 2025.3.13__py3-none-any.whl → 2025.3.14__py3-none-any.whl

owlplanner/plan.py

@@ -266,6 +266,9 @@ class Plan(object):
         # self.horizons = [yobs[i] + expectancy[i] - thisyear + 1 for i in range(self.N_i)]
         self.N_n = np.max(self.horizons)
         self.year_n = np.linspace(thisyear, thisyear + self.N_n - 1, self.N_n, dtype=np.int32)
+        # Year in the plan (if any) where individuals turn 59. For 10% withdrawal penalty.
+        self.n59 = 59 - thisyear + self.yobs
+        self.n59[self.n59 < 0] = 0
         # Handle passing of one spouse before the other.
         if self.N_i == 2 and np.min(self.horizons) != np.max(self.horizons):
             self.n_d = np.min(self.horizons)
@@ -1289,6 +1292,11 @@ class Plan(object):
             for t in range(Nt):
                 row.addElem(_q2(CF, t, n, Nt, Nn), self.theta_tn[t, n])
 
+            # Minus 10% penalty on early withdrawals.
+            if n < self.n59[i]:
+                row.addElem(_q3(Cw, i, 1, n, Ni, Nj, Nn), 0.1)
+                row.addElem(_q3(Cw, i, 2, n, Ni, Nj, Nn), 0.1)
+
             A.addRow(row, rhs, rhs)
 
         # Impose income profile.
@@ -1970,6 +1978,12 @@ class Plan(object):
         self.G_n = np.sum(self.F_tn, axis=0)
         self.T_tn = self.F_tn * self.theta_tn
         self.T_n = np.sum(self.T_tn, axis=0)
+        self.penalty_n = np.zeros(Nn)
+        # Add early withdrawal penalties if any.
+        for i in range(Ni):
+            self.penalty_n[0:self.n59[i]] += 0.1*(self.w_ijn[i, 1, 0:self.n59[i]] + self.w_ijn[i, 2, 0:self.n59[i]])
+
+        self.T_n += self.penalty_n
 
         tau_0 = np.array(self.tau_kn[0, :])
         tau_0[tau_0 < 0] = 0
@@ -2112,6 +2126,11 @@ class Plan(object):
             dic[f"-- Subtotal in tax bracket {tname}"] = f"{u.d(taxPaidNow)}"
             dic[f"-- [Subtotal in tax bracket {tname}]"] = f"{u.d(taxPaid)}"
 
+        penaltyPaid = np.sum(self.penalty_n, axis=0)
+        penaltyPaidNow = np.sum(self.penalty_n / self.gamma_n[:-1], axis=0)
+        dic["-- Subtotal in early withdrawal penalty"] = f"{u.d(penaltyPaidNow)}"
+        dic["-- [Subtotal in early withdrawal penalty]"] = f"{u.d(penaltyPaid)}"
+
         taxPaid = np.sum(self.U_n, axis=0)
         taxPaidNow = np.sum(self.U_n / self.gamma_n[:-1], axis=0)
         dic["Total tax paid on gains and dividends"] = f"{u.d(taxPaidNow)}"
@@ -2507,6 +2526,7 @@ class Plan(object):
 
         title = self._name + "\nRaw Income Sources"
         stypes = self.sources_in.keys()
+        # stypes = [item for item in stypes if "RothX" not in item]
 
         if tag != "":
             title += " - " + tag
@@ -2517,7 +2537,7 @@ class Plan(object):
         else:
             yformat = "\\$k (" + str(self.year_n[0]) + "\\$)"
             sources_in = {}
-            for key in self.sources_in:
+            for key in stypes:
                 sources_in[key] = self.sources_in[key] / self.gamma_n[:-1]
 
         fig, ax = _stackPlot(

owlplanner/utils.py

@@ -70,8 +70,8 @@ def getUnits(units) -> int:
     return fac
 
 
-# Could be a one-line lambda function:
-# krond = lambda a, b: 1 if a == b else 0
+# Next two functins could be a one-line lambda functions.
+# e.g., krond = lambda a, b: 1 if a == b else 0
 def krond(a, b) -> int:
     """
     Kronecker integer delta function.
@@ -79,6 +79,13 @@ def krond(a, b) -> int:
     return 1 if a == b else 0
 
 
+def heavyside(x) -> int:
+    """
+    Heavyside step function.
+    """
+    return 1 if x >= 0 else 0
+
+
 def roundCents(values, decimals=2):
     """
     Round values in NumPy array down to second decimal.

owlplanner/version.py

@@ -1 +1 @@
-__version__ = "2025.03.13"
+__version__ = "2025.03.14"

{owlplanner-2025.3.13.dist-info → owlplanner-2025.3.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: owlplanner
-Version: 2025.3.13
+Version: 2025.3.14
 Summary: Owl: Retirement planner with great wisdom
 Project-URL: HomePage, https://github.com/mdlacasse/owl
 Project-URL: Repository, https://github.com/mdlacasse/owl

{owlplanner-2025.3.13.dist-info → owlplanner-2025.3.14.dist-info}/RECORD

@@ -2,16 +2,16 @@ owlplanner/__init__.py,sha256=QqrdT0Qks20osBTg7h0vJHAxpP9lL7DA99xb0nYbtw4,254
 owlplanner/abcapi.py,sha256=LbzW_KcNy0IeHp42MUHwGu_H67B2h_e1_vu-c2ACTkQ,6646
 owlplanner/config.py,sha256=uJ6jZ9Rx2CB2P5cFRscsUCXOW6uml7I4pta2ozjW0uo,12263
 owlplanner/logging.py,sha256=tYMw04O-XYSzjTj36fmKJGLcE1VkK6k6oJNeqtKXzuc,2530
-owlplanner/plan.py,sha256=BjNzeV9inGGmirMTQ8VVfOaxR8-WGLZAgSATyJk0dAI,115239
+owlplanner/plan.py,sha256=2uS_N121p-fPhCboVSAthdt_uVw109ltLTvCLeGouuo,116263
 owlplanner/progress.py,sha256=8jlCvvtgDI89zXVNMBg1-lnEyhpPvKQS2X5oAIpoOVQ,384
 owlplanner/rates.py,sha256=TN407qU4n-bac1oymkQ_n2QKEPwFQxy6JZVGwgIkLQU,15585
 owlplanner/tax2025.py,sha256=B-A5eU3wxdcAaxRCbT3qI-JEKoD_ZeNbg_86XhNdQEI,7745
 owlplanner/timelists.py,sha256=tYieZU67FT6TCcQQis36JaXGI7dT6NqD7RvdEjgJL4M,4026
-owlplanner/utils.py,sha256=HM70W60qB41zfnbl2LltNwAuLYHyy5XYbwnbNcaa6FE,2351
-owlplanner/version.py,sha256=GR2k_WJJsQtJYtt5VJ64fZchrCUdIeleskErLMVRk8s,28
+owlplanner/utils.py,sha256=WpJgn79YZfH8UCkcmhd-AZlxlGuz1i1-UDBRXImsY6I,2485
+owlplanner/version.py,sha256=gPOv802nGWM84uQsfLVWHJKlgbpuozzBAjyW-OgDkzE,28
 owlplanner/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 owlplanner/data/rates.csv,sha256=6fxg56BVVORrj9wJlUGFdGXKvOX5r7CSca8uhUbbuIU,3734
-owlplanner-2025.3.13.dist-info/METADATA,sha256=k1I0OLgv8oQG7oxVARAVWXz7tWNNeynGPcB3uYndEUA,53801
-owlplanner-2025.3.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-owlplanner-2025.3.13.dist-info/licenses/LICENSE,sha256=IwGE9guuL-ryRPEKi6wFPI_zOhg7zDZbTYuHbSt_SAk,35823
-owlplanner-2025.3.13.dist-info/RECORD,,
+owlplanner-2025.3.14.dist-info/METADATA,sha256=C5BlpAPu2CwtVQmuBUPDTYi7mXg86T3b0dYR5anPUak,53801
+owlplanner-2025.3.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+owlplanner-2025.3.14.dist-info/licenses/LICENSE,sha256=IwGE9guuL-ryRPEKi6wFPI_zOhg7zDZbTYuHbSt_SAk,35823
+owlplanner-2025.3.14.dist-info/RECORD,,