From 014ca82b183248a7848c8f3190d24289ef7a5c24 Mon Sep 17 00:00:00 2001 From: stevenhua0320 Date: Thu, 1 Aug 2024 12:55:55 +0800 Subject: [PATCH] fix import modules, py2->py3 --- src/diffpy/srmise/peakextraction.py | 16 ++++++++-------- src/diffpy/srmise/peakstability.py | 3 ++- src/diffpy/srmise/srmiselog.py | 1 - 3 files changed, 10 insertions(+), 10 deletions(-) diff --git a/src/diffpy/srmise/peakextraction.py b/src/diffpy/srmise/peakextraction.py index 7ac50bc..c62102b 100644 --- a/src/diffpy/srmise/peakextraction.py +++ b/src/diffpy/srmise/peakextraction.py @@ -21,10 +21,10 @@ import numpy as np from diffpy.srmise import srmiselog -from diffpy.srmise.baselines import Baseline +from diffpy.srmise.baselines.base import Baseline from diffpy.srmise.dataclusters import DataClusters from diffpy.srmise.modelcluster import ModelCluster, ModelCovariance -from diffpy.srmise.peaks import Peak, Peaks +from diffpy.srmise.peaks.base import Peak, Peaks from diffpy.srmise.srmiseerrors import SrMiseDataFormatError, SrMiseEstimationError, SrMiseFileError logger = logging.getLogger("diffpy.srmise") @@ -131,7 +131,7 @@ def setvars(self, quiet=False, **kwds): initial_peaks: Peaks instance. These peaks are present at the start of extraction. rng: Sequence specifying the least and greatest x-values over which to extract peaks. """ - for k, v in kwds.iteritems(): + for k, v in kwds.items(): if k in self.extractvars: if quiet: logger.debug("Setting variable %s=%s", k, v) @@ -183,7 +183,7 @@ def defaultvars(self, *args): self.effective_dy = self.effective_dy * np.ones(len(self.x)) if self.pf is None or "pf" in args: - from diffpy.srmise.peaks import GaussianOverR + from diffpy.srmise.peaks.gaussianoverr import GaussianOverR # TODO: Make a more useful default. self.pf = [GaussianOverR(self.x[-1] - self.x[0])] @@ -208,13 +208,13 @@ def defaultvars(self, *args): self.baseline = None if self.baseline is None or "baseline" in args: - from diffpy.srmise.baselines import Polynomial + from diffpy.srmise.baselines.polynomial import Polynomial bl = Polynomial(degree=-1) self.baseline = bl.actualize(np.array([]), "internal") if self.error_method is None or "error_method" in args: - from diffpy.srmise.modelevaluators import AIC + from diffpy.srmise.modelevaluators.aic import AIC self.error_method = AIC @@ -1307,8 +1307,8 @@ def fit_single(self): from numpy.random import randn - from diffpy.srmise.modelevaluators import AICc - from diffpy.srmise.peaks import GaussianOverR + from diffpy.srmise.modelevaluators.aicc import AICc + from diffpy.srmise.peaks.gaussianoverr import GaussianOverR srmiselog.setlevel("info") srmiselog.liveplotting(False) diff --git a/src/diffpy/srmise/peakstability.py b/src/diffpy/srmise/peakstability.py index f53eed2..83d2897 100644 --- a/src/diffpy/srmise/peakstability.py +++ b/src/diffpy/srmise/peakstability.py @@ -15,7 +15,8 @@ import matplotlib.pyplot as plt import numpy as np -from diffpy.srmise import ModelCluster, PDFPeakExtraction +from diffpy.srmise.modelcluster import ModelCluster +from diffpy.srmise.pdfpeakextraction import PDFPeakExtraction # This is a total hack-job right now, and isn't suitable for diff --git a/src/diffpy/srmise/srmiselog.py b/src/diffpy/srmise/srmiselog.py index 361ea4b..d7b15cd 100644 --- a/src/diffpy/srmise/srmiselog.py +++ b/src/diffpy/srmise/srmiselog.py @@ -247,7 +247,6 @@ def read(self, filename): err, ) raise SrMiseFileError(emsg) - return None def readstr(self, datastring): """Read tracer ModelCluster from string.