Skip to content
This repository has been archived by the owner on May 1, 2023. It is now read-only.

Commit

Permalink
fix: config input
Browse files Browse the repository at this point in the history
  • Loading branch information
bokajgd committed Mar 6, 2023
1 parent 139e90a commit 62493fe
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion src/psycop_model_training/config_schemas/eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ class EvalConfSchema(BaseModel):
force: bool = False
# Whether to force evaluation even if wandb is not "run". Used for testing.

table_1: bool = True
descriptive_stats_table: bool = True
# Whether to generate table 1.

top_n_feature_importances: int
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -183,9 +183,9 @@ def get_descriptive_stats_table_artifact(self):
"""Returns descriptive stats table artifact."""
return [
ArtifactContainer(
label="table_1",
label="descriptive_stats_table",
artifact=DescriptiveStatsTable(
self.eval_ds
self.eval_ds,
).generate_descriptive_stats_table(),
),
]
Expand Down Expand Up @@ -225,8 +225,8 @@ def get_all_artifacts(self) -> list[ArtifactContainer]:
"""Generates artifacts from an EvalDataset."""
artifact_containers = self.create_base_plot_artifacts()

if self.cfg.eval.table_1:
artifact_containers += self.get_table_1_artifact()
if self.cfg.eval.descriptive_stats_table:
artifact_containers += self.get_descriptive_stats_table_artifact()

if self.pipe_metadata and self.pipe_metadata.feature_importances:
artifact_containers += self.get_feature_importance_artifacts()
Expand Down

0 comments on commit 62493fe

Please sign in to comment.