From fd6fdd753cf7be6c8e737eaeb6151e772034adc1 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Fri, 10 Feb 2017 18:00:30 -0500
Subject: [PATCH] remove spurious absolute mhcflurry import
---
mhcflurry/class1_allele_specific/scoring.py | 5 ++---
mhcflurry/class1_allele_specific/train.py | 4 +---
2 files changed, 3 insertions(+), 6 deletions(-)
diff --git a/mhcflurry/class1_allele_specific/scoring.py b/mhcflurry/class1_allele_specific/scoring.py
index e50ba83a..485d4303 100644
--- a/mhcflurry/class1_allele_specific/scoring.py
+++ b/mhcflurry/class1_allele_specific/scoring.py
@@ -7,7 +7,7 @@ import sklearn
import numpy
import scipy
-import mhcflurry
+from ..regression_target import ic50_to_regression_target
def make_scores(
@@ -38,8 +38,7 @@ def make_scores(
dict with entries "auc", "f1", "tau"
"""
- y_pred = mhcflurry.regression_target.ic50_to_regression_target(
- ic50_y_pred, max_ic50)
+ y_pred = ic50_to_regression_target(ic50_y_pred, max_ic50)
try:
auc = sklearn.metrics.roc_auc_score(
ic50_y <= threshold_nm,
diff --git a/mhcflurry/class1_allele_specific/train.py b/mhcflurry/class1_allele_specific/train.py
index 7e5824b8..2c6d6519 100644
--- a/mhcflurry/class1_allele_specific/train.py
+++ b/mhcflurry/class1_allele_specific/train.py
@@ -26,8 +26,6 @@ import math
import numpy
import pandas
-import mhcflurry
-
from .scoring import make_scores
from .class1_binding_predictor import Class1BindingPredictor
from ..hyperparameters import HyperparameterDefaults
@@ -186,7 +184,7 @@ def train_and_test_one_model_one_fold(
impute,
model_description))
- predictor = mhcflurry.Class1BindingPredictor(
+ predictor = Class1BindingPredictor(
max_ic50=max_ic50,
**model_params)
--
GitLab