Use yaml.safe_load to load config

zooniverse · Jan 31, 2024 · 3dcb599 · 3dcb599
1 parent 7c1ecb1
commit 3dcb599
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/process.py b/process.py
@@ -9,7 +9,7 @@
 BATCH_SIZE = 50
 
 with open('/run/secrets/config.yml', 'r') as f:
-    CONFIG = yaml.load(f)
+    CONFIG = yaml.safe_load(f)
 
 session = boto3.Session(profile_name='default')
 s3 = session.resource('s3')