[Bug] TypeError in train_dineof.py: fit() argument mismatch

[tbwang-clound]

I encountered a TypeError when running train_dineof.py. The error indicates that model.fit() was called with too many arguments.

Run the script train_dineof.py.

Traceback (most recent call last): File "train_dineof.py", line ..., in <module> model.fit(x, tmp_data) TypeError: fit() takes 2 positional arguments but 3 were given

In train_dineof.py, the code calls:

model.fit(x, tmp_data) # Passes 2 arguments (excluding self)

in model/dineof.py ：
def _fit(self, mat):
if mat.ndim > 2:
mat = rectify_tensor(mat)

    if self.to_center:
        mat, *means = center_mat(mat)