Skip to content

Commit b0c9ef1

Browse files
authored
Update contraqctor to render to optionally render to html (#452)
1 parent c3f9f70 commit b0c9ef1

File tree

3 files changed

+15
-6
lines changed

3 files changed

+15
-6
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ Changelog = "https://github.com/AllenNeuralDynamics/Aind.Behavior.VrForaging/rel
2828

2929
[project.optional-dependencies]
3030

31-
data = ["contraqctor<0.6.0"]
31+
data = ["contraqctor>=0.5.3, <0.6.0"]
3232

3333
launcher = [
3434
"aind-clabe[aind-services] >0.8 ,<0.10.0",

src/aind_behavior_vr_foraging/data_qc.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -285,11 +285,19 @@ class DataQcCli(pydantic_settings.BaseSettings, cli_kebab_case=True):
285285
description="Path to the session data directory."
286286
)
287287
version: str = pydantic.Field(default=__semver__, description="Version of the dataset.")
288+
report_path: Path | None = pydantic.Field(
289+
default=None, description="Path to save the Html QC report. If not provided, report is not saved."
290+
)
288291

289292
def cli_cmd(self):
290293
vr_dataset = dataset(Path(self.data_path), self.version)
291294
runner = make_qc_runner(vr_dataset)
292-
runner.run_all_with_progress()
295+
results = runner.run_all_with_progress()
296+
if report_path := self.report_path:
297+
from contraqctor.qc.reporters import HtmlReporter
298+
299+
reporter = HtmlReporter(output_path=report_path)
300+
reporter.report_results(results, serialize_context_exportable_obj=True)
293301

294302

295303
if __name__ == "__main__":

uv.lock

Lines changed: 5 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)