Add mode to overwrite table if already exists

This commit is contained in:
Chang She
2023-04-19 16:19:18 -07:00
parent ec197b1855
commit d7c5793803
4 changed files with 49 additions and 5 deletions

View File

@@ -156,10 +156,10 @@ class LanceTable:
return LanceQueryBuilder(self, query)
@classmethod
def create(cls, db, name, data, schema=None):
def create(cls, db, name, data, schema=None, mode="create"):
tbl = LanceTable(db, name)
data = _sanitize_data(data, schema)
lance.write_dataset(data, tbl._dataset_uri, mode="create")
lance.write_dataset(data, tbl._dataset_uri, mode=mode)
return tbl