make names in max_num_pool_nodes abit less specific

zooniverse · Dec 13, 2024 · ab82fb9 · ab82fb9
1 parent d938268
commit ab82fb9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/bajor/env_helpers.py b/bajor/env_helpers.py
@@ -16,9 +16,9 @@ def port():
     return os.environ.get('PORT', '8000')
 
 def max_num_pool_nodes(pool_id):
-    if(pool_id == 'training_1'):
+    if('training' in pool_id):
         return os.environ.get('MAX_NODES_TRAINING', 2)
-    elif(pool_id == 'predictions_0'):
+    elif('predictions' in pool_id):
         return os.environ.get('MAX_NODES_PREDICTION', 2)
     else:
         return 0