Add a pickle-size regression test for fitted learners (v1.0 M1)

Aperta
#1,063 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub

@techi101 ci sta già lavorando.

Dal 7/11/2026.

  • #1064 di @techi101 — aperta

Valutazione

Difficoltà
3/5
Tempo stimato
1-2 giorni
Idoneità per principianti
67/100
Tipo di issue
Funzionalità
Chiarezza
Abbastanza chiara
Stato di attività
Attiva
Stack tecnologico
python

Direzione di ricerca

Read the v1.0 M1 roadmap item in #938 and inspect the small datasets in conftest. Add one test file covering fitted meta-learners, causal and uplift trees, and forests by pickling each model and checking its size against the proposed budget of about twice today's size. Done means the test passes in the normal suite and catches pickle sizes that grow with training rows.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Descrizione

The v1.0 roadmap in #938 lists "a pickle-size regression guard in CI" under M1. I'd like to pick this up.

The plan is one test file that fits each meta-learner (S, T, X, R, DR, regressor and classifier) plus the causal and uplift tree and forest on the small conftest datasets, pickles each one, and checks the size against a budget of about 2x today's size. Today these pickle to roughly 1 to 20 KB and don't grow with the number of training rows, so a change that starts keeping training data on a fitted model would fail the test right away. It runs in the normal test suite in about 10 seconds with no CI changes.

Happy to adjust the learner list or headroom if you have a preference.

Lingua principale
Python
Stelle
6k
Fork
892
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Preparare l'ambiente

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Altre issue di uber/causalml

Tutte le issue di uber/causalml

Issue simili

Altre issue su Python

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.