edit target_method

app.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
-from typing import Dict
+from typing import Dict, List
 import uvicorn
 import torch
 
@@ -22,15 +22,13 @@ model.to(device)
 class SourceCode(BaseModel):
     class_name: str
     code: str
-
 class TestCase(BaseModel):
     id: str
     test_fixture: str
     name: str
     code: str
     target_class: str
-    target_method: str
-
+    target_method: List[str]
 class SimilarityInput(BaseModel):
     pair_id: str
     source_code: SourceCode