Skip to content

Commit 19515eb

Browse files
author
Rama Vasudevan
committed
adjusted_fitter_refactor_tests
1 parent e88d7f9 commit 19515eb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/proc/test_fitter_refactor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -299,6 +299,7 @@ def test_beps_fit(self):
299299

300300
from sidpy.proc.fitter_refactor import SidpyFitterRefactor
301301
import SciFiReaders as sr
302+
import numpy as np
302303

303304
from urllib.request import urlretrieve
304305

@@ -328,7 +329,7 @@ def test_beps_fit(self):
328329
fitter.setup_calc()
329330

330331
log.info("Testing Guess Function in test_beps_fit")
331-
beps_guess = fitter.do_guess().compute()
332+
beps_guess = fitter.do_guess()
332333

333334
#option 1: No Prior Fitting (Better for clean data)
334335
log.info('Testing Fitting without K-Means in test_beps_fit')
@@ -340,7 +341,6 @@ def test_beps_fit(self):
340341
result_beps = fitter.do_fit(use_kmeans=True, n_clusters=12)
341342

342343
#Check to see that the results metadata is beign written correctly
343-
import numpy as np
344344
model_source = result_beps.metadata['source_code']['model_function']
345345

346346
from scipy.special import erf

0 commit comments

Comments
 (0)