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",
|