Skip to content

Commit 3ce631e

Browse files
authored
remove HGLM from GLM python test (#5752)
1 parent be42375 commit 3ce631e

File tree

2 files changed

+0
-6
lines changed

2 files changed

+0
-6
lines changed

ml/src/test/scala/ai/h2o/sparkling/ml/models/MOJOParameterTestSuite.scala

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,8 +78,6 @@ class MOJOParameterTestSuite extends FunSuite with SharedH2OTestContext with Mat
7878
.setLabelCol("CAPSULE")
7979
.setSeed(1)
8080
.setFamily("binomial")
81-
// .setRandomFamily(Array("gaussian"))
82-
// .setRandomLink(Array("identity"))
8381
.setAlphaValue(Array(0.5))
8482
.setLambdaValue(Array(0.5))
8583
.setMaxIterations(30)

py/tests/unit/with_runtime_sparkling/test_glm.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -153,12 +153,8 @@ def createInitialGlmDefinitionForRandomCols():
153153
return H2OGLM(featuresCols=["x1", "x3", "x5", "x6"],
154154
labelCol="y",
155155
family="gaussian",
156-
randomFamily=["gaussian"],
157-
randomLink=["identity"],
158-
HGLM=True,
159156
calcLike=True)
160157

161-
162158
@pytest.mark.skip(reason="HGLM doesn't support MOJOs yet.")
163159
def testRandomColsArePropagatedToInternals(semiconductorDataset):
164160
semiconductorDataset = semiconductorDataset.withColumn("Device", semiconductorDataset.Device.cast("string"))

0 commit comments

Comments
 (0)