Merge pull request #65 from ebi-uniprot/dev

esm score endpoint - remaining changes
ebi-uniprot · Apr 26, 2024 · 73c32ae · 73c32ae
2 parents d19f546 + cd20cd2
commit 73c32ae
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/main/java/uk/ac/ebi/protvar/controller/PredictionController.java b/src/main/java/uk/ac/ebi/protvar/controller/PredictionController.java
@@ -9,10 +9,7 @@
 import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
-import uk.ac.ebi.protvar.model.data.ConservScore;
-import uk.ac.ebi.protvar.model.data.Foldx;
-import uk.ac.ebi.protvar.model.data.Interaction;
-import uk.ac.ebi.protvar.model.data.Pocket;
+import uk.ac.ebi.protvar.model.data.*;
 import uk.ac.ebi.protvar.repo.ProtVarDataRepo;
 import uk.ac.ebi.protvar.utils.AminoAcid;
 
@@ -129,7 +126,7 @@ public ResponseEntity<List<ESMScore>> getEsmScores(
             @Parameter(example = "Q9NUW8") @PathVariable String accession,
             @Parameter(example = "493") @PathVariable Integer position) {
 
-        List<ConservScore> scores = protVarDataRepo.getConservScores(accession, position);
+        List<ESMScore> scores = protVarDataRepo.getEsmScores(accession, position);
         return new ResponseEntity<>(scores, HttpStatus.OK);
     }
 }