notebook code doesn't match training.py code when training locally

[bzburr]

So in the "Data Processing and Learning.ipynb"

Same pictures training/testing

model, history = train.training(model, X_train, X_test, y_train, y_test, data_augmentation=True)#, callback=True, six_conv=True)

## Training in the notebook
X_train, X_test, y_train, y_test = train.get_dataset()
model, opt = train.create_model_four_conv(X_train.shape[1:])
model.compile(loss='categorical_crossentropy',
               optimizer=opt,
               metrics=['accuracy'])
model, history = train.training(model, X_train, X_test, y_train, y_test, data_augmentation=True, callback=True, six_conv=True)

​the train.training code from line 183 on "train.py"

def training(model, X_train, X_test, y_train, y_test, data_augmentation=True):

these don't match,...