type erros

Author: Jon Brenas

tests/anoph/test_pca.py

@@ -344,11 +344,16 @@ def test_pca_fit_exclude_samples(fixture, api: AnophelesPca):
 def test_pca_cohort_downsampling(fixture, api: AnophelesPca):
     # Parameters for selecting input data.
     all_sample_sets = api.sample_sets()["sample_set"].to_list()
-    sample_sets = np.random.sample(all_sample_sets, 2)
+    sample_sets = np.random.choice(all_sample_sets, size=2)
     data_params = dict(
         region=np.random.choice(api.contigs),
         sample_sets=sample_sets,
-        site_mask=np.random.choice((None,) + api.site_mask_ids),
+        site_mask=np.random.choice(
+            [
+                None,
+            ]
+            + api.site_mask_ids
+        ),
     )
 
     # Test cohort downsampling.