Convert env value to boolean explicitly (#105)

biocommons · Oct 22, 2024 · 6251f78 · 6251f78
1 parent 0806b64
commit 6251f78
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/anyvar/storage/sql_storage.py b/src/anyvar/storage/sql_storage.py
@@ -75,7 +75,8 @@ def __init__(
             os.environ.get("ANYVAR_SQL_STORE_BATCH_LIMIT", "100000")
         )
         self.flush_on_batchctx_exit = (
-            bool(os.environ.get("ANYVAR_SQL_STORE_FLUSH_ON_BATCHCTX_EXIT", "True"))
+            os.environ.get("ANYVAR_SQL_STORE_FLUSH_ON_BATCHCTX_EXIT", "true").lower()
+            in ["true", "yes", "1"]
             if flush_on_batchctx_exit is None
             else flush_on_batchctx_exit
         )