Skip to content

[FEAT] Implements sample_weight usage in distributed learning API#558

Merged
nasaul merged 1 commit intoNixtla:mainfrom
simonez-tuidi:feat/weight_column_distributed
Jan 29, 2026
Merged

[FEAT] Implements sample_weight usage in distributed learning API#558
nasaul merged 1 commit intoNixtla:mainfrom
simonez-tuidi:feat/weight_column_distributed

Conversation

@simonez-tuidi
Copy link
Copy Markdown
Contributor

Description

Adds support to include sample weights through the weight_col parameter in the relevant methods of DistributedMLForecast (fit, cross_validation).

Does not support Ray at this time, only Spark and Dask.

Passes existing and new tests that are included in tests/test_distributed_forecast.py

Checklist:

  • This PR has a meaningful title and a clear description.
  • The tests pass.
  • All linting tasks pass.
  • The notebooks are clean.

@CLAassistant
Copy link
Copy Markdown

CLAassistant commented Jan 29, 2026

CLA assistant check
All committers have signed the CLA.

@codspeed-hq
Copy link
Copy Markdown

codspeed-hq Bot commented Jan 29, 2026

CodSpeed Performance Report

Merging this PR will not alter performance

Comparing simonez-tuidi:feat/weight_column_distributed (b276598) with main (9bda122)

Summary

✅ 12 untouched benchmarks

@nasaul nasaul enabled auto-merge (squash) January 29, 2026 23:36
Copy link
Copy Markdown
Contributor

@nasaul nasaul left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Great job, thanks for your contribution!

@nasaul nasaul merged commit a0104a0 into Nixtla:main Jan 29, 2026
28 of 30 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants