ImportError: cannot import name 'solvent' from partially initialized module 'api.views' in django rest api

Я использую Django rest framework для предсказания двух строк и использую celery для выполнения вывода. Но я получаю ошибку при попытке взять solute и solvent из функции результата в файл tasks.py, но получаю ошибку ImportError: cannot import name 'solvent' from partially initialized module 'api.views'

Мой views.py файл

from .tasks import predictions

@api_view(['GET'])
def result(request):
    response = {}
    solute = request.GET.get('solute')
    solvent = request.GET.get('solvent')
    results = predictions.delay()
    response["interaction_map"] = (results[1].detach().numpy()).tolist()
    response["predictions"] = results[0].item()
    return Response({'result': response}, status=200)

Мой tasks.py файл

@shared_task(bind=True)
def predictions(self):

    mol = Chem.MolFromSmiles(solute)
    mol = Chem.AddHs(mol)
    solute = Chem.MolToSmiles(mol)
    solute_graph = get_graph_from_smile(solute)

    mol = Chem.MolFromSmiles(solvent)
    mol = Chem.AddHs(mol)
    solvent = Chem.MolToSmiles(mol)
    solvent_graph = get_graph_from_smile(solvent)


    delta_g, interaction_map =  model([solute_graph.to(device), solvent_graph.to(device)])
    return delta_g, torch.trunc(interaction_map)

Вернуться на верх