voly 0.0.60__py3-none-any.whl → 0.0.62__py3-none-any.whl

voly/client.py

@@ -319,7 +319,7 @@ class VolyClient:
     # -------------------------------------------------------------------------
 
     @staticmethod
-    def fit_model(market_data: pd.DataFrame,
+    def fit_model(option_chain: pd.DataFrame,
                   model_name: str = 'svi',
                   initial_params: Optional[List[float]] = None,
                   param_bounds: Optional[Tuple] = None) -> Dict[str, Any]:
@@ -327,19 +327,19 @@ class VolyClient:
         Fit a volatility model to market data.
 
         Parameters:
-        - market_data: DataFrame with market data
+        - option_chain: DataFrame with option market data
         - model_name: Name of model to fit (default: 'svi')
         - initial_params: Optional initial parameters for optimization
         - param_bounds: Optional parameter bounds for optimization
 
         Returns:
-        - Dictionary with fitting results and optional plots
+        - Tuple of (fit_results, fit_performance)
         """
         logger.info(f"Fitting {model_name.upper()} model to market data")
 
         # Fit the model
         fit_results, fit_performance = fit_model(
-            market_data=market_data,
+            option_chain=option_chain,
             model_name=model_name,
             initial_params=initial_params,
             param_bounds=param_bounds
@@ -377,7 +377,7 @@ class VolyClient:
 
     @staticmethod
     def plot_model(fit_results: Dict[str, Any],
-                   market_data: pd.DataFrame = None,
+                   option_chain: pd.DataFrame = None,
                    moneyness_params: Tuple[float, float, int] = (-2, 2, 500)
                    ) -> Dict[str, go.Figure]:
         """
@@ -385,7 +385,7 @@ class VolyClient:
 
         Parameters:
         - fit_results: Dictionary with fitting results from fit_model()
-        - market_data: Optional market data for comparison
+        - option_chain: Optional market data for comparison
         - moneyness_params: Grid of log-moneyness values
 
         Returns:
@@ -401,7 +401,7 @@ class VolyClient:
         fit_performance = fit_results['fit_performance']
 
         # Plot volatility smiles
-        plots['smiles'] = plot_all_smiles(moneyness_array, iv_surface, market_data)
+        plots['smiles'] = plot_all_smiles(moneyness_array, iv_surface, option_chain)
 
         # Plot 3D surface
         plots['surface_3d'] = plot_3d_surface(moneyness_array, iv_surface)

voly/core/charts.py

@@ -21,7 +21,7 @@ pio.renderers.default = "browser"
 @catch_exception
 def plot_volatility_smile(moneyness_array: np.ndarray,
                           iv_array: np.ndarray,
-                          market_data: pd.DataFrame = None,
+                          option_chain: pd.DataFrame = None,
                           maturity: Optional[str] = None) -> go.Figure:
     """Plot volatility smile for a single expiry."""
     fig = go.Figure()
@@ -38,8 +38,8 @@ def plot_volatility_smile(moneyness_array: np.ndarray,
     )
 
     # Add market data if provided
-    if market_data is not None and maturity is not None:
-        maturity_data = market_data[market_data['maturity_name'] == maturity]
+    if option_chain is not None and maturity is not None:
+        maturity_data = option_chain[option_chain['maturity_name'] == maturity]
 
         if not maturity_data.empty:
             # Add bid and ask IVs
@@ -72,13 +72,13 @@ def plot_volatility_smile(moneyness_array: np.ndarray,
 @catch_exception
 def plot_all_smiles(moneyness_array: np.ndarray,
                     iv_surface: Dict[str, np.ndarray],
-                    market_data: Optional[pd.DataFrame] = None) -> List[go.Figure]:
+                    option_chain: Optional[pd.DataFrame] = None) -> List[go.Figure]:
     """Plot volatility smiles for all expiries."""
     return [
         plot_volatility_smile(
             moneyness_array=moneyness_array,
             iv_array=iv_surface[maturity],
-            market_data=market_data,
+            option_chain=option_chain,
             maturity=maturity
         )
         for maturity in iv_surface.keys()

voly/core/fit.py

@@ -19,17 +19,17 @@ warnings.filterwarnings("ignore")
 
 
 @catch_exception
-def calculate_residuals(params: List[float], ytm: float, market_data: pd.DataFrame,
+def calculate_residuals(params: List[float], ytm: float, option_chain: pd.DataFrame,
                         model: Any = SVIModel) -> np.ndarray:
     """Calculate residuals between market and model implied volatilities."""
-    maturity_data = market_data[market_data['ytm'] == ytm]
+    maturity_data = option_chain[option_chain['ytm'] == ytm]
     w = np.array([model.svi(x, *params) for x in maturity_data['log_moneyness']])
     iv_actual = maturity_data['mark_iv'].values
     return iv_actual - np.sqrt(w / ytm)
 
 
 @catch_exception
-def fit_model(market_data: pd.DataFrame,
+def fit_model(option_chain: pd.DataFrame,
               model_name: str = 'svi',
               initial_params: Optional[List[float]] = None,
               param_bounds: Optional[Tuple] = None) -> Tuple[pd.DataFrame, pd.DataFrame]:
@@ -56,8 +56,8 @@ def fit_model(market_data: pd.DataFrame,
                          'rmse', 'mae', 'r2', 'max_error', 'n_points']
 
     # Get unique maturities and sort them
-    unique_ytms = sorted(market_data['ytm'].unique())
-    maturity_names = [market_data[market_data['ytm'] == ytm]['maturity_name'].iloc[0] for ytm in unique_ytms]
+    unique_ytms = sorted(option_chain['ytm'].unique())
+    maturity_names = [option_chain[option_chain['ytm'] == ytm]['maturity_name'].iloc[0] for ytm in unique_ytms]
 
     # Create empty DataFrames
     fit_results_df = pd.DataFrame(index=results_index, columns=maturity_names)
@@ -68,7 +68,7 @@ def fit_model(market_data: pd.DataFrame,
 
     for ytm in unique_ytms:
         # Get data for this maturity
-        maturity_data = market_data[market_data['ytm'] == ytm]
+        maturity_data = option_chain[option_chain['ytm'] == ytm]
         maturity_name = maturity_data['maturity_name'].iloc[0]
         dtm = maturity_data['dtm'].iloc[0]
 
@@ -79,7 +79,7 @@ def fit_model(market_data: pd.DataFrame,
             result = least_squares(
                 calculate_residuals,
                 initial_params,
-                args=(ytm, market_data, SVIModel),
+                args=(ytm, option_chain, SVIModel),
                 bounds=param_bounds,
                 max_nfev=1000
             )

{voly-0.0.60.dist-info → voly-0.0.62.dist-info}/METADATA

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

{voly-0.0.60.dist-info → voly-0.0.62.dist-info}/RECORD

@@ -1,18 +1,18 @@
 voly/__init__.py,sha256=8xyDk7rFCn_MOD5hxuv5cxxKZvBVRiSIM7TgaMPpwpw,211
-voly/client.py,sha256=I-GF9rwfUSY3_UOUqjE_r1ozwTpULFyFxt1YlDxsHvw,20454
+voly/client.py,sha256=5_duFk_9B7z6U7yADyB5W2morahDZkZf_KkEVFIWCA0,20457
 voly/exceptions.py,sha256=PBsbn1vNMvKcCJwwJ4lBO6glD85jo1h2qiEmD7ArAjs,92
 voly/formulas.py,sha256=Xgaq4lx1fNzRfu9W84fMNeH6GRJ0FNFNUUUYn5ffjjE,8843
 voly/models.py,sha256=LXXIlpXZQEfXTuCngxC8Hd3bWtw6wdXDCSGxTLmHM-c,3659
 voly/core/__init__.py,sha256=bu6fS2I1Pj9fPPnl-zY3L7NqrZSY5Zy6NY2uMUvdhKs,183
-voly/core/charts.py,sha256=kWjd3sfvTalfuElt7vSuRPtYuWpMoJ9-L2Z9Mj-mOMQ,24374
+voly/core/charts.py,sha256=0fYD748wHUBtuOcRufzfDwjlhmLf6g081KH-zquIj0A,24381
 voly/core/data.py,sha256=e8qBArubNqPkrfuIYB_q2WhRf7TKzA4Z3FhMC-xyLEE,8862
-voly/core/fit.py,sha256=YClTiODlH9CBAqv5VNBlE0k6qD22Hwic91PnTWstyj4,6490
+voly/core/fit.py,sha256=zn8o58yc83QyP9DG9JjTG0s0h2CAhYhZYRwjhT83V7w,6500
 voly/core/interpolate.py,sha256=ztVIePJZOh-CIbn69wkh1JW2rKywNe2FEewRN0zcSAo,8185
 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.60.dist-info/LICENSE,sha256=wcHIVbE12jfcBOai_wqBKY6xvNQU5E909xL1zZNq_2Q,1065
-voly-0.0.60.dist-info/METADATA,sha256=KfzS5D-9x2-pW5divEfn_aQLKlsFn0i42Hl2ewzfg7Q,4092
-voly-0.0.60.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-voly-0.0.60.dist-info/top_level.txt,sha256=ZfLw2sSxF-LrKAkgGjOmeTcw6_gD-30zvtdEY5W4B7c,5
-voly-0.0.60.dist-info/RECORD,,
+voly-0.0.62.dist-info/LICENSE,sha256=wcHIVbE12jfcBOai_wqBKY6xvNQU5E909xL1zZNq_2Q,1065
+voly-0.0.62.dist-info/METADATA,sha256=3fFRqBZffusUSlSbf3Z74pbXpNBAiu2ASZ0qBd5wz2w,4092
+voly-0.0.62.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+voly-0.0.62.dist-info/top_level.txt,sha256=ZfLw2sSxF-LrKAkgGjOmeTcw6_gD-30zvtdEY5W4B7c,5
+voly-0.0.62.dist-info/RECORD,,