Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
·
5a8f398
1
Parent(s):
72c9292
refactor schema retrieval in database setup to use collect_schema method
Browse files
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.
|
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",
|