Skip to content

Commit f412ee8

Browse files
committed
Simplify logic in factory functions with constrained (enum) inputs
1 parent 5dc5f54 commit f412ee8

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

include/stochtree/leaf_model.h

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -979,10 +979,8 @@ static inline SuffStatVariant suffStatFactory(ModelType model_type, int basis_di
979979
return createSuffStat<GaussianUnivariateRegressionSuffStat>();
980980
} else if (model_type == kMultivariateRegressionLeafGaussian) {
981981
return createSuffStat<GaussianMultivariateRegressionSuffStat, int>(basis_dim);
982-
} else if (model_type == kLogLinearVariance) {
983-
return createSuffStat<LogLinearVarianceSuffStat>();
984982
} else {
985-
Log::Fatal("Incompatible model type provided to suff stat factory");
983+
return createSuffStat<LogLinearVarianceSuffStat>();
986984
}
987985
}
988986

@@ -1002,10 +1000,8 @@ static inline LeafModelVariant leafModelFactory(ModelType model_type, double tau
10021000
return createLeafModel<GaussianUnivariateRegressionLeafModel, double>(tau);
10031001
} else if (model_type == kMultivariateRegressionLeafGaussian) {
10041002
return createLeafModel<GaussianMultivariateRegressionLeafModel, Eigen::MatrixXd>(Sigma0);
1005-
} else if (model_type == kLogLinearVariance) {
1006-
return createLeafModel<LogLinearVarianceLeafModel, double, double>(a, b);
10071003
} else {
1008-
Log::Fatal("Incompatible model type provided to leaf model factory");
1004+
return createLeafModel<LogLinearVarianceLeafModel, double, double>(a, b);
10091005
}
10101006
}
10111007

0 commit comments

Comments
 (0)