update v0.1.5

ramiz khaled · ramiz khaled · commit 936d0521ec7b · 2026-04-13T16:35:52.000+02:00
diff --git 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
@@ -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]
 

Original file line number	Diff line number	Diff line change
`@@ -63,8 +63,8 @@ def predict_amn(`
`63`	`63`	`}`
`64`	`64`	`}`
`65`	`65`
`66`		`- if organism not in PRESETS:`
`67`		`- raise ValueError(f"Organism must be one of {list(PRESETS.keys())}")`
	`66`	`+ #if organism not in PRESETS:`
	`67`	`+ #raise ValueError(f"Organism must be one of {list(PRESETS.keys())}")`
`68`	`68`
`69`	`69`	`cfg = PRESETS[organism]`
`70`	`70`
Original file line number	Diff line number	Diff line change
`@@ -61,8 +61,8 @@ def build_training_set(`
`61`	`61`	`"cobra_file": get_default_model('biolog','build')`
`62`	`62`	`}`
`63`	`63`	`}`
`64`		`- if organism not in PRESETS:`
`65`		`- raise ValueError(f"Organism must be one of {list(PRESETS.keys())}")`
	`64`	`+ #if organism not in PRESETS:`
	`65`	`+ #raise ValueError(f"Organism must be one of {list(PRESETS.keys())}")`
`66`	`66`
`67`	`67`	`cfg = PRESETS[organism]`
`68`	`68`
`@@ -149,8 +149,8 @@ def train_gr_prediction(`
`149`	`149`	`"batch_size": 100`
`150`	`150`	`}`
`151`	`151`	`}`
`152`		`- if organism not in PRESETS:`
`153`		`- raise ValueError(f"Organism must be one of {list(PRESETS.keys())}")`
	`152`	`+ #if organism not in PRESETS:`
	`153`	`+ #raise ValueError(f"Organism must be one of {list(PRESETS.keys())}")`
`154`	`154`
`155`	`155`	`cfg = PRESETS[organism]`
`156`	`156`