Allow to set AWS buckets in config

gyorilab · Aug 16, 2024 · b5c468e · b5c468e
1 parent c50ad84
commit b5c468e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/depmap_analysis/util/aws.py b/depmap_analysis/util/aws.py
@@ -5,16 +5,17 @@
 from typing import Union, Tuple, Any, Dict, Optional
 from operator import itemgetter
 
+from indra.config import get_config
 from indra.util.aws import get_s3_file_tree, get_s3_client
 from indra_db.cli.dump import Sif, StatementHashMeshId
 
 dumpers = [Sif]
 
 logger = logging.getLogger(__name__)
 
-DUMPS_BUCKET = 'bigmech'
+DUMPS_BUCKET = get_config("DUMPS_BUCKET") or 'bigmech'
 DUMPS_PREFIX = 'indra-db/dumps/'
-NET_BUCKET = 'depmap-analysis'
+NET_BUCKET = get_config("NET_BUCKET") or 'depmap-analysis'
 NETS_PREFIX = 'graphs/'