voly 0.0.38__tar.gz → 0.0.40__tar.gz

{voly-0.0.38/src/voly.egg-info → voly-0.0.40}/PKG-INFO

@@ -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 → voly-0.0.40}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "voly"
-version = "0.0.38"
+version = "0.0.40"
 description = "Options & volatility research package"
 readme = "README.md"
 authors = [
@@ -60,7 +60,7 @@ line_length = 100
 multi_line_output = 3
 
 [tool.mypy]
-python_version = "0.0.38"
+python_version = "0.0.40"
 warn_return_any = true
 warn_unused_configs = true
 disallow_untyped_defs = true

{voly-0.0.38 → voly-0.0.40}/src/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-0.0.38 → voly-0.0.40}/src/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-0.0.38 → voly-0.0.40}/src/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 → voly-0.0.40/src/voly.egg-info}/PKG-INFO

@@ -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