Code

Use narwhals for DataFrame logic#100

Code

Merged

kklein merged 57 commits intomainQuantco/metalearners:mainfrom narwhalsQuantco/metalearners:narwhalsCopy head branch name to clipboardMar 8, 2025

Commits

Commits on Feb 5, 2025

• Fix type annotation.

  kklein

  committed
  32ebeac

  Copy full SHA for 32ebeac

• Use narwhals for indexing utils.

  kklein

  committed
  5b3a419

  Copy full SHA for 5b3a419

• Return early.

  kklein

  committed
  2a78f0a

  Copy full SHA for 2a78f0a

• Use narwhals for subselection of columns.

  kklein

  committed
  b7be471

  Copy full SHA for b7be471

• Use narwhals for conversion of treatment assignments.

  kklein

  committed
  91328e1

  Copy full SHA for 91328e1

• Use narwhals in slearner.

  kklein

  committed
  01d1f4f

  Copy full SHA for 01d1f4f

• Use narwhals in treatment conversion.

  kklein

  committed
  c0365cd

  Copy full SHA for c0365cd

• Add polars as test dependency.

  kklein

  committed
  04b949d

  Copy full SHA for 04b949d

• Add polars to append_treatment test.

  kklein

  committed
  480006c

  Copy full SHA for 480006c

• Don't convert column name types.

  kklein

  committed
  621d662

  Copy full SHA for 621d662

Commits on Feb 6, 2025

• Add polars backend to test_feature_set.

  kklein

  committed
  d86cc2a

  Copy full SHA for d86cc2a

• Handle ints feature set lookup.

  kklein

  committed
  c5ef4d0

  Copy full SHA for c5ef4d0

• Test indexing against polars.

  kklein

  committed
  e4c4d07

  Copy full SHA for e4c4d07

• Test fixed_binary_propensity against polars.

  kklein

  committed
  5733dbc

  Copy full SHA for 5733dbc

• Add polars backend to test_crossfitestimator_oos_smoke.

  kklein

  committed
  c4f15cf

  Copy full SHA for c4f15cf

• Add polars backend to test_validate_outcome_* tests.

  kklein

  committed
  83dd478

  Copy full SHA for 83dd478

• Fix docstring.

  kklein

  committed
  30483c2

  Copy full SHA for 30483c2

• Add polars backend to test_evaluate_feature_set_smoke.

  kklein

  committed
  ade0b9c

  Copy full SHA for ade0b9c

• Properly filter vectors if boolean mask is given.

  kklein

  committed
  942a3de

  Copy full SHA for 942a3de

• Subselect with index helper.

  kklein

  committed
  ccda67b

  Copy full SHA for ccda67b

• Add test case for boolean masking.

  kklein

  committed
  7d39bb5

  Copy full SHA for 7d39bb5

• Add test case for boolean masking.

  kklein

  committed
  8626100

  Copy full SHA for 8626100

• Support boolean masking for matrices.

  kklein

  committed
  d14e063

  Copy full SHA for d14e063

• Adapt learners to handle non-np, non-pd input datastructures.

  kklein

  committed
  5567c17

  Copy full SHA for 5567c17

Commits on Feb 13, 2025

• Merge branch 'main' of github.com:Quantco/metalearners into narwhals

  kklein

  committed
  e0ccd75

  Copy full SHA for e0ccd75

• Remove redundant tolist invocations.

  kklein

  committed
  faec207

  Copy full SHA for faec207

• Use nth for collection of ints.

  kklein

  committed
  dd9c07e

  Copy full SHA for dd9c07e

• Bring back necessary tolist call.

  kklein

  committed
  306a792

  Copy full SHA for 306a792

• Fix test.

  kklein

  committed
  284d595

  Copy full SHA for 284d595

Commits on Feb 15, 2025

• Merge branch 'main' of github.com:Quantco/metalearners into narwhals

  kklein

  committed
  3ae892a

  Copy full SHA for 3ae892a

Commits on Feb 18, 2025

• Always specify eager_only=True in nw.from_native.

  kklein

  committed
  38cfb4a

  Copy full SHA for 38cfb4a

• Adapt comments.

  kklein

  committed
  2021a38

  Copy full SHA for 2021a38

• Use stable api.

  kklein

  committed
  3d108ee

  Copy full SHA for 3d108ee

• Address TODO.

  kklein

  committed
  1fae16f

  Copy full SHA for 1fae16f

Commits on Feb 19, 2025

• Add TODO.

  kklein

  committed
  89bd519

  Copy full SHA for 89bd519

• Fix typo.

  kklein

  committed
  3bd2ad6

  Copy full SHA for 3bd2ad6

• Use polars for typing.

  kklein

  committed
  bdabcd9

  Copy full SHA for bdabcd9

• Don't ignore typing.

  kklein

  committed
  3d42b97

  Copy full SHA for 3d42b97

• Add more tests against polars.

  kklein

  committed
  50b025f

  Copy full SHA for 50b025f

• Fix import.

  kklein

  committed
  8885d7a

  Copy full SHA for 8885d7a

Commits on Feb 24, 2025

• Merge branch 'main' of github.com:Quantco/metalearners into narwhals

  kklein

  committed
  e7fb130

  Copy full SHA for e7fb130

• Fix typo.

  kklein

  committed
  2d87255

  Copy full SHA for 2d87255

Commits on Feb 26, 2025

• Rework handling of categories in S-Learner. (#105)

  Show description for 186f099
  kklein

  authored
  186f099

  Copy full SHA for 186f099

• Remove dead breakpoint.

  kklein

  committed
  47a7eab

  Copy full SHA for 47a7eab

• Enable mypy.

  kklein

  committed
  ba0ac26

  Copy full SHA for ba0ac26

• Include csr_matrix in check for matrix.

  kklein

  committed
  3ad4f5b

  Copy full SHA for 3ad4f5b

• Update metalearners/_narwhals_utils.py

  Show description for 5e4b5b7
  

  kklein

  and

  FBruzzesi

  authored
  5e4b5b7

  Copy full SHA for 5e4b5b7

• Update metalearners/_narwhals_utils.py

  Show description for 740a5a3
  

  kklein

  and

  FBruzzesi

  authored
  740a5a3

  Copy full SHA for 740a5a3

• Apply pchs.

  kklein

  committed
  f37f7e3

  Copy full SHA for f37f7e3

• Check for null values in nw one-hot-encoding.

  kklein

  committed
  f535c40

  Copy full SHA for f535c40

• Merge branch 'main' of github.com:Quantco/metalearners into narwhals

  kklein

  committed
  cbcd50e

  Copy full SHA for cbcd50e

• Use md instead of rst in docstrings.

  kklein

  committed
  0c0f415

  Copy full SHA for 0c0f415

Commits on Feb 27, 2025

• Update metalearners/_narwhals_utils.py

  Show description for ae7f613
  

  kklein

  and

  FBruzzesi

  authored
  ae7f613

  Copy full SHA for ae7f613

• Hotfix issue regarding interaction between lightgbm and polars.

  kklein

  committed
  f7dc89d

  Copy full SHA for f7dc89d

Commits on Mar 8, 2025

• Add todo.

  kklein

  committed
  8721b5c

  Copy full SHA for 8721b5c

• Update lock.

  kklein

  committed
  0e595fc

  Copy full SHA for 0e595fc

• Fix test.

  kklein

  committed
  e95897f

  Copy full SHA for e95897f