Skip to content

Commit 80a95b5

Browse files
authored
Fix build break (#146)
The previous 2 changes conflicted. Resovling the break that happened between them.
1 parent 33c8364 commit 80a95b5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/Microsoft.ML/Runtime/EntryPoints/MacroUtils.cs

+2-2
Original file line numberDiff line numberDiff line change
@@ -197,10 +197,10 @@ public static T TrainerKindApiValue<T>(TrainerKinds trainerKind)
197197

198198
public static bool IsTrainerOfKind(Type type, TrainerKinds trainerKind)
199199
{
200-
if (type == typeof(Trainers.BinaryLogisticRegressor))
200+
if (type == typeof(Trainers.LogisticRegressionBinaryClassifier))
201201
return trainerKind == TrainerKinds.SignatureBinaryClassifierTrainer;
202202

203-
if (type == typeof(Trainers.LogisticRegressor))
203+
if (type == typeof(Trainers.LogisticRegressionClassifier))
204204
return trainerKind == TrainerKinds.SignatureMultiClassClassifierTrainer;
205205

206206
if (trainerKind != TrainerKinds.SignatureMultiClassClassifierTrainer && trainerKind != TrainerKinds.SignatureMultiOutputRegressorTrainer)

0 commit comments

Comments
 (0)