davanstrien HF Staff commited on
Commit
5a8f398
·
1 Parent(s): 72c9292

refactor schema retrieval in database setup to use collect_schema method

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -200,7 +200,7 @@ def setup_database():
200
  logger.info(f"Row count of new model data: {model_row_count}")
201
 
202
  if model_collection.count() < model_row_count:
203
- schema = model_lazy_df.schema
204
  select_columns = [
205
  "modelId",
206
  "summary",
 
200
  logger.info(f"Row count of new model data: {model_row_count}")
201
 
202
  if model_collection.count() < model_row_count:
203
+ schema = model_lazy_df.collect_schema()
204
  select_columns = [
205
  "modelId",
206
  "summary",