TYP: add schedule (#541)

Co-authored-by: JHM Darbyshire (win11) <[email protected]>

pyproject.toml

@@ -140,7 +140,8 @@ ignore = [
 files = [
     "python/rateslib/calendars/**/*.py",
     "python/rateslib/fx/**/*.py",
-    "python/rateslib/dual/**/*.py"
+    "python/rateslib/dual/**/*.py",
+    "python/rateslib/scheduling.py"
 ]
 strict = true
 

python/rateslib/default.py

@@ -129,7 +129,7 @@ def __init__(self):
             "T": 4,
             "S": 6,
             "A": 12,
-            "Z": 1e8,
+            "Z": 120000,  # corresponds to 10,000 years
         }
         self.eom = False
         self.eom_fx = True
@@ -380,9 +380,9 @@ def plot3d(x, y, z, labels=None):
     return fig, ax, None
 
 
-def _drb(default: Any, possible_blank: Any | NoInput) -> Any:
+def _drb(default: Any, possible_ni: Any | NoInput) -> Any:
     """(D)efault (r)eplaces (b)lank"""
-    return default if possible_blank is NoInput.blank else possible_blank
+    return default if isinstance(possible_ni, NoInput) else possible_ni
 
 
 def _make_py_json(json: str, class_name: str) -> str:

python/rateslib/dual/__init__.py

@@ -3,6 +3,7 @@
 import math
 from functools import partial
 from statistics import NormalDist
+from typing import TypeAlias
 
 import numpy as np
 
@@ -12,8 +13,8 @@
 Dual.__doc__ = "Dual number data type to perform first derivative automatic differentiation."
 Dual2.__doc__ = "Dual number data type to perform second derivative automatic differentiation."
 
-DualTypes = float | Dual | Dual2 | Variable
-Number = float | Dual | Dual2
+DualTypes: TypeAlias = "float | Dual | Dual2 | Variable"
+Number: TypeAlias = "float | Dual | Dual2"
 
 # Licence: Creative Commons - Attribution-NonCommercial-NoDerivatives 4.0 International
 # Commercial use of this code, and/or copying and redistribution is prohibited.