From 309b00ba0b8648ee679892d04a17af1870bcdeea Mon Sep 17 00:00:00 2001 From: Amrest Chinkamol Date: Wed, 15 Nov 2023 06:50:53 +0000 Subject: [PATCH] fix: Add context manager before enter .array_exists --- tiledb/dataframe_.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tiledb/dataframe_.py b/tiledb/dataframe_.py index c41a7918cc..da72af9ef1 100644 --- a/tiledb/dataframe_.py +++ b/tiledb/dataframe_.py @@ -865,8 +865,9 @@ def from_csv(uri: str, csv_file: Union[str, List[str]], **kwargs): elif mode not in ["ingest", "append"]: raise tiledb.TileDBError("Invalid mode specified ('{}')".format(mode)) - if mode != "append" and tiledb.array_exists(uri): - raise tiledb.TileDBError("Array URI '{}' already exists!".format(uri)) + with tiledb.scope_ctx(ctx): + if mode != "append" and tiledb.array_exists(uri): + raise tiledb.TileDBError("Array URI '{}' already exists!".format(uri)) # this is a pandas pass-through argument, do not pop! chunksize = kwargs.get("chunksize", None)