Method __init__

tests/test_deepspeed_training_worker.py:23–29

(self, model, optimizer)

class FakeDeepSpeedEngine:
    def __init__(self, model, optimizer):
        self.model = model
        self.optimizer = optimizer
        self.forward_calls = 0
        self.backward_calls = 0
        self.step_calls = 0
        self.zero_grad_calls = 0

    def __call__(self, *args, **kwargs):
        self.forward_calls += 1