Skip to content

Commit f75fd20

Browse files
committed
Pass weights=None to _fit_gmm_density calls
Explicitly sets weights=None in calls to _fit_gmm_density for both GPI GMM and joint GMM fitting. This clarifies that no sample weights are used during GMM fitting.
1 parent a8a7548 commit f75fd20

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/non_local_detector/likelihoods/clusterless_gmm.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -355,6 +355,7 @@ def fit_clusterless_gmm_encoding_model(
355355
# GPI GMM (position only)
356356
gpi_gmm = _fit_gmm_density(
357357
X=enc_pos,
358+
weights=None,
358359
n_components=gmm_components_gpi,
359360
random_state=gmm_random_state,
360361
covariance_type=gmm_covariance_type_gpi,
@@ -365,6 +366,7 @@ def fit_clusterless_gmm_encoding_model(
365366
joint_samples = jnp.concatenate([enc_pos, elect_feats], axis=1)
366367
joint_gmm = _fit_gmm_density(
367368
X=joint_samples,
369+
weights=None,
368370
n_components=gmm_components_joint,
369371
random_state=gmm_random_state,
370372
covariance_type=gmm_covariance_type_joint,

0 commit comments

Comments
 (0)