Add lazy option to cfa.stf.data functions

[Copilot]

• Add lazy: bool = True parameter to get_nhsn_hrd in get_data.py
• Add lazy: bool = True parameter to get_nssp in get_data.py
• Update return type annotations in get_nhsn_hrd and get_nssp to pl.DataFrame | pl.LazyFrame
• Add @overload decorators to get_nhsn_hrd and get_nssp for Literal[True] → pl.LazyFrame and Literal[False] → pl.DataFrame inference
• Add lazy: bool = True parameter to _filter_param_estimates in get_nnh_pmfs.py
• Propagate lazy to _filter_param_estimates in all three public NNH functions
• Run linter to verify changes

[damonbayer]

@copilot Use the @overload decorator so that we can infer whether the result will be a pl.LazyFrame or pl.DataFrame.

[Copilot]

@copilot Use the @overload decorator so that we can infer whether the result will be a pl.LazyFrame or pl.DataFrame.

Done in 172da56. Added @overload signatures to both get_nhsn_hrd and get_nssp — lazy: Literal[True] resolves to pl.LazyFrame and lazy: Literal[False] resolves to pl.DataFrame.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 59.92%. Comparing base (3a556ef) to head (38a6690).
⚠️ Report is 2 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #1029   +/-   ##
=======================================
  Coverage   59.92%   59.92%           
=======================================
  Files          31       31           
  Lines        2041     2041           
=======================================
  Hits         1223     1223           
  Misses        818      818           

Flag	Coverage Δ
epiautogp	73.18% <ø> (ø)
hewr	76.98% <ø> (ø)
pipelines	56.37% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[Copilot]

Pull request overview

This PR adds an opt-in/opt-out “lazy Polars” mode to STF data-access functions so callers can choose between returning pl.DataFrame (eager) vs pl.LazyFrame (lazy) and get better type inference via overloads.

Changes:

• Added lazy: bool = True parameter and Literal-based overloads to get_nhsn_hrd and get_nssp, updating return annotations to pl.DataFrame | pl.LazyFrame.
• Updated NNH PMF helpers to accept pl.LazyFrame and introduced overloads for _filter_param_estimates.
• Introduced output = "pl_lazy" if lazy else "pl" paths when loading via datacat.

Reviewed changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 6 comments.

File	Description
cfa/stf/data/get_data.py	Adds lazy option + overloads to NHSN HRD and NSSP getters and adjusts datacat loading accordingly.
cfa/stf/data/get_nnh_pmfs.py	Extends PMF extraction/filtering to work with LazyFrame and adds overloads for the filtering helper.

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.