From 420d8b89ff93074dea818a90bd27268308d1d099 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Tue, 28 Feb 2017 21:29:35 -0500
Subject: [PATCH] fix test
---
.../class1_allele_specific_ensemble/measurement_collection.py | 1 +
test/test_ensemble.py | 4 ++--
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/mhcflurry/class1_allele_specific_ensemble/measurement_collection.py b/mhcflurry/class1_allele_specific_ensemble/measurement_collection.py
index 72e16767..6e7a06cf 100644
--- a/mhcflurry/class1_allele_specific_ensemble/measurement_collection.py
+++ b/mhcflurry/class1_allele_specific_ensemble/measurement_collection.py
@@ -179,6 +179,7 @@ class MeasurementCollection(object):
assert len(self.df) > 0
dataset = self.to_dataset(include_ms=False)
+ assert len(dataset) > 0
imputer = imputer_from_name(impute_method, **imputer_args)
result_df = dataset.impute_missing_values(
imputation_method=imputer,
diff --git a/test/test_ensemble.py b/test/test_ensemble.py
index 77d9c7b4..099aeaae 100644
--- a/test/test_ensemble.py
+++ b/test/test_ensemble.py
@@ -99,8 +99,8 @@ def test_train_command():
dropout_probability=[.25],
n_training_epochs=[10],
imputer_args=[{"n_burn_in": 2, "n_imputations": 10}],
- impute_min_observations_per_peptide=[5],
- impute_min_observations_per_allele=[2000], # limit num alleles
+ impute_min_observations_per_peptide=[1],
+ impute_min_observations_per_allele=[1],
)
print("Model selection will be over %d models" % len(models))
--
GitLab