Skip to content

Commit 6089a8a

Browse files
committed
Fix test
1 parent 9c0584e commit 6089a8a

1 file changed

Lines changed: 2 additions & 7 deletions

File tree

h2o-algos/src/test/java/hex/glm/GLMControlVariablesAndRemoveOffsetTest.java

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2410,13 +2410,8 @@ public void testRoMojoPojoGuard() {
24102410
glm = new GLM(params).trainModel().get();
24112411
Scope.track_generic(glm);
24122412

2413-
// With remove_offset_effects=true, the exported MOJO/POJO would score WITH offset
2414-
// but the H2O model scores WITHOUT offset. haveMojo()/havePojo() should return false.
2415-
// Currently haveMojo() only checks control_variables but NOT remove_offset_effects.
2416-
assertFalse("haveMojo() should return false when remove_offset_effects=true " +
2417-
"(MOJO scoring doesn't implement offset removal)", glm.haveMojo());
2418-
assertFalse("havePojo() should return false when remove_offset_effects=true " +
2419-
"(POJO scoring doesn't implement offset removal)", glm.havePojo());
2413+
assertTrue("haveMojo() should return true when remove_offset_effects=true ", glm.haveMojo());
2414+
assertTrue("havePojo() should return true when remove_offset_effects=true ", glm.havePojo());
24202415
} finally {
24212416
if (train != null) train.remove();
24222417
if (glm != null) glm.remove();

0 commit comments

Comments
 (0)