From 6251f7837a2502d71f7ee8cb2da60e8ac802f8dd Mon Sep 17 00:00:00 2001 From: Eugene Clark Date: Tue, 22 Oct 2024 16:05:07 -0400 Subject: [PATCH] Convert env value to boolean explicitly (#105) --- src/anyvar/storage/sql_storage.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/anyvar/storage/sql_storage.py b/src/anyvar/storage/sql_storage.py index 0732d42..c02e43a 100644 --- 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 )