voly 0.0.38__py3-none-any.whl → 0.0.40__py3-none-any.whl

voly/core/charts.py

@@ -346,13 +346,13 @@ def plot_fit_performance(fit_performance: pd.DataFrame) -> go.Figure:
 
 
 @catch_exception
-def plot_3d_surface(moneyness_grid: np.ndarray,
+def plot_3d_surface(moneyness_array: np.ndarray,
                     iv_surface: dict[float, np.ndarray]) -> go.Figure:
     """
     Plot 3D implied volatility surface.
 
     Parameters:
-    - moneyness_grid: grid of moneyness values
+    - moneyness_array: grid of moneyness values
     - iv_surface: Dictionary mapping expiry times to IV arrays
 
     Returns:
@@ -361,13 +361,13 @@ def plot_3d_surface(moneyness_grid: np.ndarray,
     start_color = '#60AEFC'
     end_color = '#002040'  # Darker blue
     custom_blue_scale = [[0, start_color], [1, end_color]]
-    yte_values = list(iv_surface.keys())
+    maturities = list(iv_surface.keys())
 
     # Convert implied volatility surface to array
-    z_array = np.array([iv_surface[t] for t in yte_values])
+    z_array = np.array([iv_surface[m] for m in maturities])
 
     # Create mesh grid
-    X, Y = np.meshgrid(moneyness_grid, yte_values)
+    X, Y = np.meshgrid(moneyness_array, maturities)
     Z = z_array * 100  # Convert to percentage
 
     # Create 3D surface plot with custom blue colorscale
@@ -393,7 +393,7 @@ def plot_3d_surface(moneyness_grid: np.ndarray,
 
 
 @catch_exception
-def plot_rnd(moneyness_grid: np.ndarray,
+def plot_rnd(moneyness_array: np.ndarray,
              rnd_values: np.ndarray,
              spot_price: float = 1.0,
              title: str = 'Risk-Neutral Density') -> go.Figure:
@@ -401,7 +401,7 @@ def plot_rnd(moneyness_grid: np.ndarray,
     Plot risk-neutral density (RND).
 
     Parameters:
-    - moneyness_grid: Grid of log-moneyness values
+    - moneyness_array: Grid of log-moneyness values
     - rnd_values: RND values
     - spot_price: Spot price for converting to absolute prices
     - title: Plot title
@@ -413,10 +413,10 @@ def plot_rnd(moneyness_grid: np.ndarray,
     fig = go.Figure()
 
     # Convert to prices and normalize RND
-    prices = spot_price * np.exp(moneyness_grid)
+    prices = spot_price * np.exp(moneyness_array)
 
     # Normalize the RND to integrate to 1
-    dx = moneyness_grid[1] - moneyness_grid[0]
+    dx = moneyness_array[1] - moneyness_array[0]
     total_density = np.sum(rnd_values) * dx
     normalized_rnd = rnd_values / total_density if total_density > 0 else rnd_values
 
@@ -463,7 +463,7 @@ def plot_rnd(moneyness_grid: np.ndarray,
 
 
 @catch_exception
-def plot_rnd_all_expiries(moneyness_grid: np.ndarray,
+def plot_rnd_all_expiries(moneyness_array: np.ndarray,
                           rnd_surface: Dict[str, np.ndarray],
                           param_matrix: pd.DataFrame,
                           spot_price: float = 1.0) -> go.Figure:
@@ -471,7 +471,7 @@ def plot_rnd_all_expiries(moneyness_grid: np.ndarray,
     Plot risk-neutral densities for all expiries.
 
     Parameters:
-    - moneyness_grid: Grid of log-moneyness values
+    - moneyness_array: Grid of log-moneyness values
     - rnd_surface: Dictionary mapping maturity names to RND arrays
     - param_matrix: Matrix containing model parameters with maturity info
     - spot_price: Spot price for converting to absolute prices
@@ -494,7 +494,7 @@ def plot_rnd_all_expiries(moneyness_grid: np.ndarray,
               for i in range(n_maturities)]
 
     # Convert to prices
-    prices = spot_price * np.exp(moneyness_grid)
+    prices = spot_price * np.exp(moneyness_array)
 
     # Add traces for each expiry
     for i, maturity_name in enumerate(maturity_names):
@@ -502,7 +502,7 @@ def plot_rnd_all_expiries(moneyness_grid: np.ndarray,
         dte = dte_values[maturity_name]
 
         # Normalize the RND
-        dx = moneyness_grid[1] - moneyness_grid[0]
+        dx = moneyness_array[1] - moneyness_array[0]
         total_density = np.sum(rnd) * dx
         normalized_rnd = rnd / total_density if total_density > 0 else rnd
 
@@ -543,7 +543,7 @@ def plot_rnd_all_expiries(moneyness_grid: np.ndarray,
 
 
 @catch_exception
-def plot_rnd_3d(moneyness_grid: np.ndarray,
+def plot_rnd_3d(moneyness_array: np.ndarray,
                 rnd_surface: Dict[str, np.ndarray],
                 param_matrix: pd.DataFrame,
                 spot_price: float = 1.0) -> go.Figure:
@@ -551,7 +551,7 @@ def plot_rnd_3d(moneyness_grid: np.ndarray,
     Plot 3D surface of risk-neutral densities.
 
     Parameters:
-    - moneyness_grid: Grid of log-moneyness values
+    - moneyness_array: Grid of log-moneyness values
     - rnd_surface: Dictionary mapping maturity names to RND arrays
     - param_matrix: Matrix containing model parameters with maturity info
     - spot_price: Spot price for converting to absolute prices
@@ -569,7 +569,7 @@ def plot_rnd_3d(moneyness_grid: np.ndarray,
     dte_list = [dte_values[name] for name in maturity_names]
 
     # Convert to prices
-    prices = spot_price * np.exp(moneyness_grid)
+    prices = spot_price * np.exp(moneyness_array)
 
     # Create z-data matrix and normalize RNDs
     z_data = np.zeros((len(maturity_names), len(prices)))
@@ -578,7 +578,7 @@ def plot_rnd_3d(moneyness_grid: np.ndarray,
         rnd = rnd_surface[name]
 
         # Normalize the RND
-        dx = moneyness_grid[1] - moneyness_grid[0]
+        dx = moneyness_array[1] - moneyness_array[0]
         total_density = np.sum(rnd) * dx
         normalized_rnd = rnd / total_density if total_density > 0 else rnd
 
@@ -770,7 +770,7 @@ def plot_rnd_statistics(rnd_statistics: pd.DataFrame,
 
 
 @catch_exception
-def plot_cdf(moneyness_grid: np.ndarray,
+def plot_cdf(moneyness_array: np.ndarray,
              rnd_values: np.ndarray,
              spot_price: float = 1.0,
              title: str = 'Cumulative Distribution Function') -> go.Figure:
@@ -778,7 +778,7 @@ def plot_cdf(moneyness_grid: np.ndarray,
     Plot the cumulative distribution function (CDF) from RND values.
 
     Parameters:
-    - moneyness_grid: Grid of log-moneyness values
+    - moneyness_array: Grid of log-moneyness values
     - rnd_values: RND values
     - spot_price: Spot price for converting to absolute prices
     - title: Plot title
@@ -787,10 +787,10 @@ def plot_cdf(moneyness_grid: np.ndarray,
     - Plotly figure
     """
     # Convert to prices and normalize RND
-    prices = spot_price * np.exp(moneyness_grid)
+    prices = spot_price * np.exp(moneyness_array)
 
     # Normalize the RND
-    dx = moneyness_grid[1] - moneyness_grid[0]
+    dx = moneyness_array[1] - moneyness_array[0]
     total_density = np.sum(rnd_values) * dx
     normalized_rnd = rnd_values / total_density if total_density > 0 else rnd_values
 
@@ -850,7 +850,7 @@ def plot_cdf(moneyness_grid: np.ndarray,
 
 
 @catch_exception
-def plot_pdf(moneyness_grid: np.ndarray,
+def plot_pdf(moneyness_array: np.ndarray,
              rnd_values: np.ndarray,
              spot_price: float = 1.0,
              title: str = 'Probability Density Function') -> go.Figure:
@@ -858,7 +858,7 @@ def plot_pdf(moneyness_grid: np.ndarray,
     Plot the probability density function (PDF) from RND values.
 
     Parameters:
-    - moneyness_grid: Grid of log-moneyness values
+    - moneyness_array: Grid of log-moneyness values
     - rnd_values: RND values
     - spot_price: Spot price for converting to absolute prices
     - title: Plot title
@@ -867,7 +867,7 @@ def plot_pdf(moneyness_grid: np.ndarray,
     - Plotly figure
     """
     # This is essentially the same as plot_rnd but with a different title
-    return plot_rnd(moneyness_grid, rnd_values, spot_price, title)
+    return plot_rnd(moneyness_array, rnd_values, spot_price, title)
 
 
 @catch_exception
@@ -886,13 +886,13 @@ def plot_interpolated_surface(
     - Plotly figure
     """
     # Extract data from interpolation results
-    moneyness_grid = interp_results['moneyness_grid']
+    moneyness_array = interp_results['moneyness_array']
     target_expiries_years = interp_results['target_expiries_years']
     iv_surface = interp_results['iv_surface']
 
     # Create a 3D surface plot
     fig = plot_3d_surface(
-        moneyness=moneyness_grid,
+        moneyness=moneyness_array,
         expiries=target_expiries_years,
         iv_surface=iv_surface,
         title=title

voly/core/fit.py

@@ -295,6 +295,6 @@ def get_iv_surface(fit_results: Dict[str, Any],
     for maturity, yte in zip(maturity_values, yte_values):
         svi_params = param_matrix[maturity].values
         w_svi = [SVIModel.svi(x, *svi_params) for x in moneyness_array]
-        iv_surface[yte] = np.sqrt(np.array(w_svi) / yte)
+        iv_surface[maturity] = np.sqrt(np.array(w_svi) / yte)
 
     return moneyness_array, iv_surface

voly/core/rnd.py

@@ -43,11 +43,9 @@ def get_rnd_surface(fit_results: Dict[str, Any],
     maturity_values = fit_results['fit_performance']['Maturity']
     param_matrix = fit_results['raw_param_matrix']
 
-    # Generate implied volatility for each expiry
+    # Generate rnd for each expiry
     for maturity, yte in zip(maturity_values, yte_values):
         svi_params_list = list(param_matrix[maturity].values)
-
-        # Adjust parameter order to match function signatures
         a, b, sigma, rho, m = svi_params_list
 
         # Calculate total variance
@@ -55,8 +53,7 @@ def get_rnd_surface(fit_results: Dict[str, Any],
 
         # Calculate risk-neutral density using the base RND function
         rnd_values = np.array([rnd(x, var) for x, var in zip(moneyness_array, total_var)])
-
-        rnd_surface[maturity_name] = rnd_values
+        rnd_surface[maturity] = rnd_values
 
     return moneyness_array, rnd_surface
 

{voly-0.0.38.dist-info → voly-0.0.40.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: voly
-Version: 0.0.38
+Version: 0.0.40
 Summary: Options & volatility research package
 Author-email: Manu de Cara <manu.de.cara@gmail.com>
 License: MIT

{voly-0.0.38.dist-info → voly-0.0.40.dist-info}/RECORD

@@ -4,15 +4,15 @@ voly/exceptions.py,sha256=PBsbn1vNMvKcCJwwJ4lBO6glD85jo1h2qiEmD7ArAjs,92
 voly/formulas.py,sha256=7NQZtYWFT_VOyAVNQHLB5-8caGAdtzU8y8eFcYJsZn8,7256
 voly/models.py,sha256=YJ12aamLz_-aOni4Qm0_XV9u4bjKK3vfJz0J2gc1h0o,3565
 voly/core/__init__.py,sha256=bu6fS2I1Pj9fPPnl-zY3L7NqrZSY5Zy6NY2uMUvdhKs,183
-voly/core/charts.py,sha256=lHiDYKb4uHK2R7wjs8xuxQ6FD9QsQyFR2-6bbtxqquE,26696
+voly/core/charts.py,sha256=tt8WZfAKmfb5YQiVXYAXt9ROT_FRMB4w_Lbe5ihjh5E,26725
 voly/core/data.py,sha256=Dfk-ByHpdteUiLXr0p-wRLr3VAmdyjdDBKwjwdTgCjA,9939
-voly/core/fit.py,sha256=LVu_bgYJppH9Sboh1ovQv5BnJnCxSyNSX7lCpv0NnYk,9945
+voly/core/fit.py,sha256=O4PMihVWI1NIEFU4_RkKvT73p0Jk0tB-ot5moMXAW78,9950
 voly/core/interpolate.py,sha256=ztVIePJZOh-CIbn69wkh1JW2rKywNe2FEewRN0zcSAo,8185
-voly/core/rnd.py,sha256=0jiBkDDy6d_AQggCZsmkd1ilWp4wzAoeCGn_rKybsSA,11897
+voly/core/rnd.py,sha256=8FTU-Qp9epW9yE4XSOdiFGIRXrGyXqF6mVgZn1NMvxk,11813
 voly/utils/__init__.py,sha256=E05mWatyC-PDOsCxQV1p5Xi1IgpOomxrNURyCx_gB-w,200
 voly/utils/logger.py,sha256=4-_2bVJmq17Q0d7Rd2mPg1AeR8gxv6EPvcmBDMFWcSM,1744
-voly-0.0.38.dist-info/LICENSE,sha256=wcHIVbE12jfcBOai_wqBKY6xvNQU5E909xL1zZNq_2Q,1065
-voly-0.0.38.dist-info/METADATA,sha256=ABlHe4U0hIrdrV9C3N9CrzvA41_SAT5FaZvt1tApp6Y,4092
-voly-0.0.38.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-voly-0.0.38.dist-info/top_level.txt,sha256=ZfLw2sSxF-LrKAkgGjOmeTcw6_gD-30zvtdEY5W4B7c,5
-voly-0.0.38.dist-info/RECORD,,
+voly-0.0.40.dist-info/LICENSE,sha256=wcHIVbE12jfcBOai_wqBKY6xvNQU5E909xL1zZNq_2Q,1065
+voly-0.0.40.dist-info/METADATA,sha256=82ZAciFjV9qvdgsfzv3L-wjRbwfjUUhUjwzzHUb0j0E,4092
+voly-0.0.40.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+voly-0.0.40.dist-info/top_level.txt,sha256=ZfLw2sSxF-LrKAkgGjOmeTcw6_gD-30zvtdEY5W4B7c,5
+voly-0.0.40.dist-info/RECORD,,