diff --git a/src/polyA_classifier.py b/src/polyA_classifier.py index 428d5b5c2f54c841fbea2e41d3eb4ba7d08a9076..4570e9d9b70b7e0bd582b2073cd61f15090672e7 100644 --- a/src/polyA_classifier.py +++ b/src/polyA_classifier.py @@ -49,7 +49,7 @@ class PolyAClassifier: 'C': 1.0 } - def __init__(self, model = Net, state_dict_path: str = './models/internal_priming.pth'): + def __init__(self, model=Net, state_dict_path: str = './models/internal_priming.pth'): """Returns a stateless classifier with the model loaded. Args: