Как создать задачу с помощью celery в django ? Проблема с POO?

Я пытаюсь установить задачу с Django и Celery. Конфигурация Celery и Django в порядке, ничего не нужно сообщать по этой части. Однако у меня есть проблема, я думаю, с написанием моего кода в ООП. Я не могу определить, где проблема. Это проблема с аргументами, 4 аргумента ожидает delay(), но мой метод ожидает только 3. Вот файлы и то, что я пытался сделать, а также трассировка ошибки.

Мои файлы

helpers.py

я хочу создать задачу для run_test_optimization method

class MachineLearniaXHelpers(MagicModels):
    def __init__(self, data_dict, target_name, type_model, test_size = 0.33, random_state = 42, **kwargs) -> None:
        super().__init__()
        self.test_size = test_size
        self.random_state = random_state
        self.process = DataProcessor(
            data=data_dict,
            target_name=target_name
        )
        self.metrics = MetricsScorer(type_model=type_model)
        self.model = self.several_algorythme(type_model, **kwargs)
    
    @staticmethod
    def split_data(data_dict):
        return train_test_split(*data_dict)
        
    @staticmethod   
    def train_model(model, X, y):
        model.fit(X, y)
    
    def run_test_optimization(self):
        dict_result = []
        for model in self.model:
            feature_model, values = self.process.transform_dict_to_array_structure()
            x_train, x_test, y_train, y_test = self.split_data(values)
            self.train_model(model, x_train, y_train)
            dict_metrics_train = self.metrics.choice_metrics_by_type_model(y_train, model.predict(x_train))
            dict_metrics_test = self.metrics.choice_metrics_by_type_model(y_test, model.predict(x_test))
            dict_result.append({
                "name_model" : model.__class__.__name__,
                "features_model" : feature_model,
                "train_performances" : dict_metrics_train,
                "test_performances" : dict_metrics_test
            })
        return dict_result

tasks.py

создайте мою задачу task_run_test_optimization

from celery import shared_task
from .helpers import MachineLearniaXHelpers

@shared_task
def task_run_test_optimization(data_dict, target_name, type_model):
    constructor = MachineLearniaXHelpers(
        data_dict,
        target_name,
        type_model
        )
    dict_result = constructor.run_test_optimization()
    return dict_result

views.py

API post метод, метод задачи здесь task_run_test_optimization

class MachineLearningXView(APIView):
    serializer_class = BuildModelMLSerializer
    permission_classes = [IsAuthenticated, UserPermissionMachineLearniaX]


    def post(self, request):
        if not self.request.session.exists(self.request.session.session_key):
            self.request.session.create()
        
        serializer = self.serializer_class(data=request.data)
        if serializer.is_valid():
            data = request.data.get('data')
            target = serializer.data.get('target_name')
            type_model = serializer.data.get('test_type')
            result = task_run_test_optimization.delay(data, target, type_model) # probleme is here
            print(result)
            return Response(status=status.HTTP_200_OK)

result send me a id in my console, the post request is ok:

System check identified no issues (0 silenced).
February 08, 2023 - 17:25:28
Django version 4.1.5, using settings 'ialab.settings'
Starting development server at http://127.0.0.1:8000/
Quit the server with CTRL-BREAK.
829e66ee-925f-4aff-8c42-49420e5758ce <-- PRINT result IS HERE
[08/Feb/2023 17:25:47] "POST /api/machinelearniax HTTP/1.1" 200 0

Но в моем celery worker, у меня такая ошибка:

[2023-02-08 17:25:47,813: INFO/MainProcess] Task api.tasks.task_run_test_optimization[829e66ee-925f-4aff-8c42-49420e5758ce] received
[2023-02-08 17:25:47,897: ERROR/MainProcess] Task api.tasks.task_run_test_optimization[829e66ee-925f-4aff-8c42-49420e5758ce] raised unexpected: TypeError('task_run_test_optimization() takes 3 positional arguments but 4 were given')
Traceback (most recent call last):
  File "C:\Users\basti\Desktop\IALab\ialab\.ialabenv\lib\site-packages\celery\app\trace.py", line 451, in trace_task
    R = retval = fun(*args, **kwargs)
  File "C:\Users\basti\Desktop\IALab\ialab\.ialabenv\lib\site-packages\celery\app\trace.py", line 734, in __protected_call__
    return self.run(*args, **kwargs)
TypeError: task_run_test_optimization() takes 3 positional arguments but 4 were given

задание получено, но не работает.

где моя ошибка?

Спасибо

Возможный случай и решение №1.

Возможно, изначально у вас была task_run_test_optimization, принимающая 4 аргумента. Вы запустили Celery, и он загрузил в память эту задачу и ее сигнатуру.

Тогда вы могли изменить сигнатуру task_run_test_optimization, чтобы она принимала 3 аргумента, но забыли перезагрузить Celery. В результате вы пытаетесь вызвать задачу с 3 аргументами, но Celery все еще хранит старую версию задачи (задачу с 4 аргументами).

Возможный случай и решение #2:

Возможно, у вас есть 2 разные задачи с одинаковым названием: "task_run_test_optimization", и ваш views.py импортирует не ту задачу? Убедитесь, что вы импортируете правильную.

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