Reference simulation function in example from top-level api

sgkit-dev · Oct 1, 2020 · 237bc96 · 237bc96
1 parent 39171ea
commit 237bc96
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sgkit/stats/pca.py b/sgkit/stats/pca.py
@@ -234,7 +234,6 @@ def pca(
     >>> import xarray as xr
     >>> import numpy as np
     >>> import sgkit as sg
-    >>> from sgkit.testing import simulate_genotype_call_dataset
 
     >>> # Set parameters for number of variants and number of samples per cohort
     >>> n_variant, n_sample = 100, [25, 50, 75]
@@ -248,7 +247,7 @@ def pca(
 
     >>>  # Run PCA on simulated dataset
     >>> ds = (
-    ...     simulate_genotype_call_dataset(n_variant=n_variant, n_sample=sum(n_sample), seed=0)
+    ...     sg.simulate_genotype_call_dataset(n_variant=n_variant, n_sample=sum(n_sample), seed=0)
     ...     .assign(call_alternate_allele_count=(("variants", "samples"), rs.binomial(2, af.T).astype("int8")))
     ...     .pipe(sg.pca, n_components=2, merge=False)
     ... )