voly 0.0.249__py3-none-any.whl → 0.0.251__py3-none-any.whl

voly/core/fit.py

@@ -67,8 +67,7 @@ class SVICalibrator:
     def filter_market_data(self, group):
         """Filter and prepare market data"""
         # Filter for call options only
-        group = group[group['flag'] == 'C']
-        group['log_moneyness'] = np.log(group['strikes'] / group['spot_price'].iloc[0])
+        group = group[group['option_type'] == 'C']
 
         # Handle duplicated IVs by keeping the row closest to log_moneyness=0
         duplicated_iv = group[group.duplicated('mark_iv', keep=False)]
@@ -150,7 +149,7 @@ class SVICalibrator:
             result = self.failed_calibration(expiry, maturity, t, len(k))
             logger.error(f'\033[31mFAILED\033[0m for {maturity} (insufficient data points)')
             self.update_results(result)
-            return maturity
+            return expiry
 
         # Perform SVI fitting
         params, loss = SVIModel.fit(tiv=w, vega=vega, k=k, tau=t)
@@ -160,10 +159,10 @@ class SVICalibrator:
             result = self.failed_calibration(expiry, maturity, t, len(k))
             logger.error(f'\033[31mFAILED\033[0m for {maturity}')
             self.update_results(result)
-            return maturity
+            return expiry
 
         # Successful fitting
-        self.params_dict[maturity] = (t, params)
+        self.params_dict[expiry] = (t, params)
 
         # Calculate all model statistics
         stats = self.calculate_model_stats(params, t, k, iv)
@@ -185,7 +184,7 @@ class SVICalibrator:
         logger.info(f'\033[32mSUCCESS\033[0m for {maturity}')
 
         self.update_results(result)
-        return maturity
+        return expiry
 
     def update_results(self, result_row):
         """Update results data dictionary"""
@@ -211,7 +210,7 @@ class SVICalibrator:
         # Create results DataFrame and mapping for updates
         fit_results = pd.DataFrame(self.results_data, index=self.results_data['maturity'])
         fit_results = fit_results.sort_values(by='t')
-        maturity_dict = {row['expiry']: idx for idx, row in fit_results.iterrows()}
+        maturities_dict = {row['expiry']: idx for idx, row in fit_results.iterrows()}
 
         # Check for calendar arbitrage
         sorted_maturities = sorted(self.params_dict.keys(), key=lambda x: self.params_dict[x][0])
@@ -221,11 +220,11 @@ class SVICalibrator:
 
         # Update calendar arbitrage status
         for mat in sorted_maturities:
-            mat_name = maturity_dict[mat]
+            mat_name = maturities_dict[mat]
             fit_results.at[mat_name, 'calendar_arbitrage_free'] = calendar_arbitrage_free
 
         # Correct calendar arbitrage violations
-        self.correct_calendar_arbitrage(sorted_maturities, fit_results, maturity_dict)
+        self.correct_calendar_arbitrage(sorted_maturities, fit_results, maturities_dict)
 
         # Clean up results and report execution time
         fit_results = fit_results.drop(columns='maturity')
@@ -234,7 +233,7 @@ class SVICalibrator:
 
         return fit_results
 
-    def correct_calendar_arbitrage(self, sorted_maturities, fit_results, maturity_dict):
+    def correct_calendar_arbitrage(self, sorted_maturities, fit_results, maturities_dict):
         """Handle calendar arbitrage corrections"""
         for i in range(1, len(sorted_maturities)):
             mat2 = sorted_maturities[i]
@@ -260,7 +259,7 @@ class SVICalibrator:
 
             # Calculate new stats and update results
             stats = self.calculate_model_stats(new_params, t2, k, iv)
-            mat2_name = maturity_dict[mat2]
+            mat2_name = maturities_dict[mat2]
 
             # Update all stats at once
             for key, value in stats.items():
@@ -274,7 +273,7 @@ class SVICalibrator:
 
         # Update final status
         for mat in sorted_maturities:
-            mat_name = maturity_dict[mat]
+            mat_name = maturities_dict[mat]
             fit_results.at[mat_name, 'calendar_arbitrage_free'] = calendar_arbitrage_free
 
 

{voly-0.0.249.dist-info → voly-0.0.251.dist-info}/METADATA

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

{voly-0.0.249.dist-info → voly-0.0.251.dist-info}/RECORD

@@ -6,15 +6,15 @@ voly/models.py,sha256=CGJQr13Uie7iwtx2hjViN9lMXeRN_uOqzp4u8NPaTlA,9282
 voly/core/__init__.py,sha256=bu6fS2I1Pj9fPPnl-zY3L7NqrZSY5Zy6NY2uMUvdhKs,183
 voly/core/charts.py,sha256=6MSU0z01fPOVSssodxdnFqchzDfupSmXq_e71WwDmVQ,12635
 voly/core/data.py,sha256=A0zrIjg-pf3mG28UR313Bc57h74G5smWCVBB7vysM_4,13527
-voly/core/fit.py,sha256=vO2wDFEk37ap8a_K0qYmk60WiUne-5AKdqHR7l0DmRc,14006
+voly/core/fit.py,sha256=diRCAqBURekXa0SYLFKb8HOT3xS611YUQvUDiwbar3c,13929
 voly/core/hd.py,sha256=dSv197RmSWFWbRRdoBzMrD_poT7ZiJ8hdD_wKE-Li_M,13559
 voly/core/interpolate.py,sha256=GAkrqaar7A0D6UMipXQUp4vSHRfb44TmCG5Xiv9elXg,5176
 voly/core/rnd.py,sha256=wiZ5OIjPDf1Th5_sQ9CZG5JgAo3EL8f63T_Rj1_VP-0,13214
 voly/utils/__init__.py,sha256=E05mWatyC-PDOsCxQV1p5Xi1IgpOomxrNURyCx_gB-w,200
 voly/utils/density.py,sha256=ONpRli-IaJDgOZ2sb27HHFc9_tkkGSATKl94JODd86A,5879
 voly/utils/logger.py,sha256=4-_2bVJmq17Q0d7Rd2mPg1AeR8gxv6EPvcmBDMFWcSM,1744
-voly-0.0.249.dist-info/licenses/LICENSE,sha256=wcHIVbE12jfcBOai_wqBKY6xvNQU5E909xL1zZNq_2Q,1065
-voly-0.0.249.dist-info/METADATA,sha256=esMpD7hJhhOij31lp-x1K78oZ0QEenrEXYx0tDirTok,4115
-voly-0.0.249.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-voly-0.0.249.dist-info/top_level.txt,sha256=ZfLw2sSxF-LrKAkgGjOmeTcw6_gD-30zvtdEY5W4B7c,5
-voly-0.0.249.dist-info/RECORD,,
+voly-0.0.251.dist-info/licenses/LICENSE,sha256=wcHIVbE12jfcBOai_wqBKY6xvNQU5E909xL1zZNq_2Q,1065
+voly-0.0.251.dist-info/METADATA,sha256=uiYM2x3Kro8MEb_TKqQsdCFQ-mZ_GPRK8siP15v_xa4,4115
+voly-0.0.251.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+voly-0.0.251.dist-info/top_level.txt,sha256=ZfLw2sSxF-LrKAkgGjOmeTcw6_gD-30zvtdEY5W4B7c,5
+voly-0.0.251.dist-info/RECORD,,