From 1f9c54e872b093bb9160d6edc3d09569dd8bca82 Mon Sep 17 00:00:00 2001 From: Kevin Johnson Date: Sun, 25 Jul 2021 13:23:58 -0700 Subject: [PATCH] ENH update mamode --- README.md | 2 +- pandas_ta/momentum/squeeze.py | 4 ++-- pandas_ta/momentum/squeeze_pro.py | 4 ++-- setup.py | 2 +- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/README.md b/README.md index c9bc563..459cca5 100644 --- a/README.md +++ b/README.md @@ -112,7 +112,7 @@ $ pip install pandas_ta Latest Version -------------- -Best choice! Version: *0.3.09b* +Best choice! Version: *0.3.10b* * Includes all fixes and updates between **pypi** and what is covered in this README. ```sh $ pip install -U git+https://github.com/twopirllc/pandas-ta diff --git a/pandas_ta/momentum/squeeze.py b/pandas_ta/momentum/squeeze.py index 514d149..2fc3ace 100644 --- a/pandas_ta/momentum/squeeze.py +++ b/pandas_ta/momentum/squeeze.py @@ -9,7 +9,7 @@ from pandas_ta.utils import get_offset from pandas_ta.utils import unsigned_differences, verify_series -def squeeze(high, low, close, bb_length=None, bb_std=None, kc_length=None, kc_scalar=None, mom_length=None, mom_smooth=None, use_tr=None, offset=None, **kwargs): +def squeeze(high, low, close, bb_length=None, bb_std=None, kc_length=None, kc_scalar=None, mom_length=None, mom_smooth=None, use_tr=None, mamode=None, offset=None, **kwargs): """Indicator: Squeeze Momentum (SQZ)""" # Validate arguments bb_length = int(bb_length) if bb_length and bb_length > 0 else 20 @@ -30,7 +30,7 @@ def squeeze(high, low, close, bb_length=None, bb_std=None, kc_length=None, kc_sc asint = kwargs.pop("asint", True) detailed = kwargs.pop("detailed", False) lazybear = kwargs.pop("lazybear", False) - mamode = kwargs.pop("mamode", "sma").lower() + mamode = mamode if isinstance(mamode, str) else "sma" def simplify_columns(df, n=3): df.columns = df.columns.str.lower() diff --git a/pandas_ta/momentum/squeeze_pro.py b/pandas_ta/momentum/squeeze_pro.py index 61cefc6..d2d591f 100644 --- a/pandas_ta/momentum/squeeze_pro.py +++ b/pandas_ta/momentum/squeeze_pro.py @@ -9,7 +9,7 @@ from pandas_ta.utils import get_offset from pandas_ta.utils import unsigned_differences, verify_series -def squeeze_pro(high, low, close, bb_length=None, bb_std=None, kc_length=None, kc_scalar_wide=None, kc_scalar_normal=None, kc_scalar_narrow=None, mom_length=None, mom_smooth=None, use_tr=None, offset=None, **kwargs): +def squeeze_pro(high, low, close, bb_length=None, bb_std=None, kc_length=None, kc_scalar_wide=None, kc_scalar_normal=None, kc_scalar_narrow=None, mom_length=None, mom_smooth=None, use_tr=None, mamode=None, offset=None, **kwargs): """Indicator: Squeeze Momentum (SQZ) PRO""" # Validate arguments bb_length = int(bb_length) if bb_length and bb_length > 0 else 20 @@ -35,7 +35,7 @@ def squeeze_pro(high, low, close, bb_length=None, bb_std=None, kc_length=None, k use_tr = kwargs.setdefault("tr", True) asint = kwargs.pop("asint", True) detailed = kwargs.pop("detailed", False) - mamode = kwargs.pop("mamode", "sma").lower() + mamode = mamode if isinstance(mamode, str) else "sma" def simplify_columns(df, n=3): df.columns = df.columns.str.lower() diff --git a/setup.py b/setup.py index a759419..a9a9e78 100644 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ setup( "pandas_ta.volatility", "pandas_ta.volume" ], - version=".".join(("0", "3", "09b")), + version=".".join(("0", "3", "10b")), description=long_description, long_description=long_description, author="Kevin Johnson",