Skip to content

Commit ddcd0f0

Browse files
fixed bug in unit test
1 parent 309800d commit ddcd0f0

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

batchglm/unit_test/test_acc_glm_all.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -306,13 +306,6 @@ def basic_test(
306306
else:
307307
raise ValueError("noise model %s not recognized" % self.noise_model)
308308

309-
estimator = _TestAccuracyGlmAllEstim(
310-
simulator=self.simulator(train_loc=train_loc),
311-
quick_scale=False if train_scale else True,
312-
noise_model=self.noise_model,
313-
sparse=sparse,
314-
init_mode=init_mode
315-
)
316309
for algo in algos:
317310
logger.info("algorithm: %s" % algo)
318311
if algo in ["ADAM", "RMSPROP", "GD"]:
@@ -348,6 +341,13 @@ def basic_test(
348341
glm.pkg_constants.JACOBIAN_MODE = "analytic"
349342
else:
350343
return ValueError("algo %s not recognized" % algo)
344+
estimator = _TestAccuracyGlmAllEstim(
345+
simulator=self.simulator(train_loc=train_loc),
346+
quick_scale=False if train_scale else True,
347+
noise_model=self.noise_model,
348+
sparse=sparse,
349+
init_mode=init_mode
350+
)
351351
estimator.estimate(
352352
algo=algo,
353353
batched=batched,

0 commit comments

Comments
 (0)