Skip to content

Commit

Permalink
Resolved naming issues in egcs #121
Browse files Browse the repository at this point in the history
  • Loading branch information
GwennyGit committed Aug 26, 2024
1 parent 58dea0c commit abc374b
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions src/refinegems/classes/egcs.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@
import warnings

from functools import partial
from itertools import product
from multiprocess import Pool
from tqdm import tqdm
from typing import Literal
Expand All @@ -43,15 +42,15 @@
"PROTON": {"Hydrogen [H(+)]": 1, "Hydrogen [H(+)] transported": -1}
} #: :meta:

ECG_SCORING_MATRIX = {'MR':1, 'RB':3, 'RF':3,'RM':6} #: :meta:
EGC_SCORING_MATRIX = {'MR':1, 'RB':3, 'RF':3,'RM':6} #: :meta:

################################################################################
# classes
################################################################################

class EGCSolver():
"""Parent class for the EGC solvers with generally useful functions and
attributes. Can only be used to find, not solve ECGs directly.
attributes. Can only be used to find, not solve EGCs directly.
Attributes:
- theshold: Float describing the cutoff, under which the model
Expand Down Expand Up @@ -371,7 +370,7 @@ class GreedyEGCSolver(EGCSolver):
Integers describing the penalty scores.
"""

def __init__(self, scoring_matrix:dict = ECG_SCORING_MATRIX) -> None:
def __init__(self, scoring_matrix:dict = EGC_SCORING_MATRIX) -> None:
super().__init__()
self.scoring_matrix = scoring_matrix

Expand Down

0 comments on commit abc374b

Please sign in to comment.