From 1f44fdd57c0d2bcbe18b40f8aa9fb32587bb5d77 Mon Sep 17 00:00:00 2001 From: Abe Levitan Date: Wed, 8 Nov 2023 13:09:10 +0100 Subject: [PATCH] Some misc work toward the no analyzer jones matrix model (polarization swept) --- src/cdtools/datasets/__init__.py | 4 ++-- src/cdtools/models/__init__.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/cdtools/datasets/__init__.py b/src/cdtools/datasets/__init__.py index ebccc20..b5ab33d 100644 --- a/src/cdtools/datasets/__init__.py +++ b/src/cdtools/datasets/__init__.py @@ -29,8 +29,8 @@ attempting to do so # I don't believe that __all__ really needed, but it's nice to define it # to be explicit that import * is safe -__all__ = ['CDataset','Ptycho2DDataset','PolarizedPtycho2DDataset'] +__all__ = ['CDataset','Ptycho2DDataset','PolarizationSwepttycho2DDataset'] from cdtools.datasets.base import CDataset from cdtools.datasets.ptycho_2d_dataset import Ptycho2DDataset -from cdtools.datasets.polarized_ptycho_2d_dataset import PolarizedPtycho2DDataset +from cdtools.datasets.polarization_swept_ptycho_2d_dataset import PolarizationSweptPtycho2DDataset diff --git a/src/cdtools/models/__init__.py b/src/cdtools/models/__init__.py index df61f96..0bc6689 100644 --- a/src/cdtools/models/__init__.py +++ b/src/cdtools/models/__init__.py @@ -22,12 +22,13 @@ defining a new ptychography model before attempting to do so. # I don't believe that __all__ really needed, but it's nice to define it # to be explicit that import * is safe -__all__ = ['CDIModel', 'SimplePtycho', 'FancyPtycho', 'PolarizedFancyPtycho', 'Bragg2DPtycho', 'Multislice2DPtycho', 'RPI', 'TimeResolvedPtychoCalibration', 'TimeResolvedRPI'] +__all__ = ['CDIModel', 'SimplePtycho', 'FancyPtycho', 'PolarizationSweptPtycho', 'PolarizedFancyPtycho', 'Bragg2DPtycho', 'Multislice2DPtycho', 'RPI', 'TimeResolvedPtychoCalibration', 'TimeResolvedRPI'] from cdtools.models.base import CDIModel from cdtools.models.simple_ptycho import SimplePtycho from cdtools.models.fancy_ptycho import FancyPtycho from cdtools.models.polarized_fancy_ptycho import PolarizedFancyPtycho +from cdtools.models.polarization_swept_ptycho import PolarizationSweptPtycho from cdtools.models.bragg_2d_ptycho import Bragg2DPtycho from cdtools.models.multislice_2d_ptycho import Multislice2DPtycho from cdtools.models.rpi import RPI