dwker/ttools
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
David Brazda
2024-10-18 12:37:33 +02:00
parent 9d70678e8c
commit 246ea48281
2 changed files with 6 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -2,7 +2,7 @@ from setuptools import setup, find_packages
setup(
name='ttools',
version='0.2.6',
version='0.2.7',
packages=find_packages(),
install_requires=[
'vectorbtpro',

17
ttools/vbtindicators.py
View File

@ -80,25 +80,18 @@ def register_custom_inds(indicator_name: str = None, if_exists: str ="skip"):
raise ValueError(f"Indicator '{indicator_name}' not found")
else:
for var_name, var_value in globals().items():
if var_name.startswith("IND_") and isinstance(var_value, vbt.IndicatorFactory):
if var_name.startswith("IND_"):
vbt.IF.register_custom_indicator(var_value, location="ttools", if_exists=if_exists)
def deregister_custom_inds(indicator_name: str = None):
"""Deregister a custom indicator or all custom indicators.
If `indicator_name` is provided, only the indicator with that name is registered.
Otherwise, all indicators are registered - they are the ones starting with "IND_" .
If `indicator_name` is provided, only the indicator with that name is deregistered.
Otherwise, all ttools indicators are deregistered.
This function does not have an `if_exists` argument.
"""
if indicator_name is not None:
var_name = f"IND_{indicator_name}"
var_value = globals().get(var_name)
if var_value is not None and isinstance(var_value, vbt.IndicatorFactory):
vbt.IF.deregister_custom_indicator(var_value, location="ttools")
else:
raise ValueError(f"Indicator '{indicator_name}' not found")
vbt.IF.deregister_custom_indicator(indicator_name, location="ttools")
else:
for var_name, var_value in globals().items():
if var_name.startswith("IND_") and isinstance(var_value, vbt.IndicatorFactory):
vbt.IF.deregister_custom_indicator(var_value, location="ttools")
vbt.IF.deregister_custom_indicator(location="ttools")