voly 0.0.241__py3-none-any.whl → 0.0.242__py3-none-any.whl

voly/formulas.py

@@ -16,35 +16,33 @@ from functools import wraps
 def vectorize_inputs(func):
     @wraps(func)
     def wrapper(*args, **kwargs):
+        # Extract input argument names from the function
         arg_names = func.__code__.co_varnames[:func.__code__.co_argcount]
         input_dict = dict(zip(arg_names, args))
         input_dict.update(kwargs)
 
-        vector_lengths = {
-            k: len(v) for k, v in input_dict.items()
-            if isinstance(v, (list, np.ndarray, pd.Series))
-        }
+        values = list(input_dict.values())
 
-        is_vectorized = len(vector_lengths) > 0
+        n_vectors = sum(isinstance(v, (list, np.ndarray, pd.Series)) for v in values)
+        n_scalars = sum(not isinstance(v, (list, np.ndarray, pd.Series)) for v in values)
 
-        if is_vectorized:
-            lengths = set(vector_lengths.values())
-            if len(lengths) != 1:
-                raise VolyError("All vectorized inputs must have the same length.")
+        if n_vectors > 0 and n_scalars > 0:
+            raise VolyError("Cannot mix scalar and vector inputs. Pass all as vectors or all as scalars.")
 
-            if any(
-                not isinstance(v, (list, np.ndarray, pd.Series))
-                for k, v in input_dict.items()
-            ):
-                raise VolyError("Cannot mix scalar and vector inputs. Pass all as vectors or all as scalars.")
+        if n_vectors > 0:
+            # All inputs are vectors, make sure they're same length
+            lengths = [len(v) for v in values]
+            if len(set(lengths)) != 1:
+                raise VolyError("All vectorized inputs must have the same length.")
 
-            max_len = next(iter(lengths))
+            # Apply vectorized logic
             return [
                 func(*[input_dict[name][i] for name in arg_names])
-                for i in range(max_len)
+                for i in range(lengths[0])
             ]
-        else:
-            return func(*args, **kwargs)
+
+        # All inputs are scalars
+        return func(*args, **kwargs)
 
     return wrapper
 

{voly-0.0.241.dist-info → voly-0.0.242.dist-info}/METADATA

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

{voly-0.0.241.dist-info → voly-0.0.242.dist-info}/RECORD

@@ -1,7 +1,7 @@
 voly/__init__.py,sha256=8xyDk7rFCn_MOD5hxuv5cxxKZvBVRiSIM7TgaMPpwpw,211
 voly/client.py,sha256=H4BY5n1isHOqMYX-Koe8-WKBKE6zxbdDYopDF58BPBI,14533
 voly/exceptions.py,sha256=PBsbn1vNMvKcCJwwJ4lBO6glD85jo1h2qiEmD7ArAjs,92
-voly/formulas.py,sha256=nSXfQ1brQAs19j2Uj2JRy-h0rXB0_EXX0FOrIU14rRQ,9953
+voly/formulas.py,sha256=YGN8cy3Wt_LX_BKSYOMpBCrRA1g39RINTNRHTXRykWM,10007
 voly/models.py,sha256=CGJQr13Uie7iwtx2hjViN9lMXeRN_uOqzp4u8NPaTlA,9282
 voly/core/__init__.py,sha256=bu6fS2I1Pj9fPPnl-zY3L7NqrZSY5Zy6NY2uMUvdhKs,183
 voly/core/charts.py,sha256=6MSU0z01fPOVSssodxdnFqchzDfupSmXq_e71WwDmVQ,12635
@@ -13,8 +13,8 @@ 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.241.dist-info/licenses/LICENSE,sha256=wcHIVbE12jfcBOai_wqBKY6xvNQU5E909xL1zZNq_2Q,1065
-voly-0.0.241.dist-info/METADATA,sha256=wWQAgMLFY-8qj7dvBoDfP2jSbHx4TsdYTkTkYREmsL0,4115
-voly-0.0.241.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-voly-0.0.241.dist-info/top_level.txt,sha256=ZfLw2sSxF-LrKAkgGjOmeTcw6_gD-30zvtdEY5W4B7c,5
-voly-0.0.241.dist-info/RECORD,,
+voly-0.0.242.dist-info/licenses/LICENSE,sha256=wcHIVbE12jfcBOai_wqBKY6xvNQU5E909xL1zZNq_2Q,1065
+voly-0.0.242.dist-info/METADATA,sha256=YXnwe-p-mZGYY8GMD-1XDGaEsFWyBxjlGRcc-2TCem4,4115
+voly-0.0.242.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+voly-0.0.242.dist-info/top_level.txt,sha256=ZfLw2sSxF-LrKAkgGjOmeTcw6_gD-30zvtdEY5W4B7c,5
+voly-0.0.242.dist-info/RECORD,,