@ -17,8 +17,8 @@ def resolve_model_by_pattern(pattern: str) -> models.Model:
return models.GeneralModel()
if pattern == 'PEAK':
return models.PeakModel()
if pattern == 'REVERSE_PEAK':
return models.ReversePeakModel()
if pattern == 'TROUGH':
return models.TroughModel()
if pattern == 'DROP':
return models.DropModel()
if pattern == 'JUMP':
@ -3,5 +3,5 @@ from models.drop_model import DropModel
from models.peak_model import PeakModel
from models.jump_model import JumpModel
from models.custom_model import CustomModel
from models.reverse_peak_model import ReversePeakModel
from models.trough_model import TroughModel
from models.general_model import GeneralModel
@ -11,7 +11,7 @@ from typing import Optional
WINDOW_SIZE = 240
class ReversePeakModel(Model):
class TroughModel(Model):
def __init__(self):
super()