Machine Learning in Elixir: Missing model in Scholar prediction (page 103 & 104)

Version: B1.0

On page 103 & 104 where the code example is:

Scholar.Linear.LogisticRegression.fit( train_inputs,
train_targets,
num_classes: 3
)
test_preds = Scholar.Linear.LogisticRegression.predict(test_inputs)

Throws an error: __ (UndefinedFunctionError) function Scholar.Linear.LogisticRegression.predict/1 is undefined or private. Did you mean: predict/2__

I believe the code should be:

model = Scholar.Linear.LogisticRegression.fit(
  train_inputs,
  train_targets,
  num_classes: 3
)
test_preds = Scholar.Linear.LogisticRegression.predict(model, test_inputs)