diff --git a/amn/_version.py b/amn/_version.py index bbab024..1276d02 100644 --- a/amn/_version.py +++ b/amn/_version.py @@ -1 +1 @@ -__version__ = "0.1.4" +__version__ = "0.1.5" diff --git a/amn/predict.py b/amn/predict.py index b690ea8..ae1681d 100644 --- a/amn/predict.py +++ b/amn/predict.py @@ -63,8 +63,8 @@ def predict_amn( } } - if organism not in PRESETS: - raise ValueError(f"Organism must be one of {list(PRESETS.keys())}") + #if organism not in PRESETS: + #raise ValueError(f"Organism must be one of {list(PRESETS.keys())}") cfg = PRESETS[organism] diff --git a/amn/train.py b/amn/train.py index 9bed295..39e07ff 100644 --- a/amn/train.py +++ b/amn/train.py @@ -61,8 +61,8 @@ def build_training_set( "cobra_file": get_default_model('biolog','build') } } - if organism not in PRESETS: - raise ValueError(f"Organism must be one of {list(PRESETS.keys())}") + #if organism not in PRESETS: + #raise ValueError(f"Organism must be one of {list(PRESETS.keys())}") cfg = PRESETS[organism] @@ -149,8 +149,8 @@ def train_gr_prediction( "batch_size": 100 } } - if organism not in PRESETS: - raise ValueError(f"Organism must be one of {list(PRESETS.keys())}") + #if organism not in PRESETS: + #raise ValueError(f"Organism must be one of {list(PRESETS.keys())}") cfg = PRESETS[organism]