Merge pull request #58 from zooniverse/zoobot-upgrade-testing

make names in max_num_pool_nodes abit less specific
zooniverse · Dec 13, 2024 · c17d99c · c17d99c
2 parents d938268 + ab82fb9
commit c17d99c
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