p4DSTEM.phase/classification style

py4dstem · Nov 13, 2023 · df069f8 · df069f8
1 parent 37da9de
commit df069f8
Showing 1 changed file with 17 additions and 15 deletions.
diff --git a/py4DSTEM/__init__.py b/py4DSTEM/__init__.py
@@ -52,8 +52,8 @@
     BraggVectorMap,
 )
 
-# TODO - classification
-# from py4DSTEM.process.classification import (
+from py4DSTEM.process import classification 
+# import (
 #     BraggVectorClassification,
 #     ConvergenceWarning,
 #     DataCube,
@@ -76,18 +76,19 @@
 
 
 # ptycho
-from py4DSTEM.process.phase import (
-    DPCReconstruction,
-    MixedstatePtychographicReconstruction,
-    MultislicePtychographicReconstruction,
-    OptimizationParameter,
-    OverlapMagneticTomographicReconstruction,
-    OverlapTomographicReconstruction,
-    ParallaxReconstruction,
-    PtychographyOptimizer,
-    SimultaneousPtychographicReconstruction,
-    SingleslicePtychographicReconstruction,
-)
+from py4DSTEM.process import phase 
+# import (
+#     DPCReconstruction,
+#     MixedstatePtychographicReconstruction,
+#     MultislicePtychographicReconstruction,
+#     OptimizationParameter,
+#     OverlapMagneticTomographicReconstruction,
+#     OverlapTomographicReconstruction,
+#     ParallaxReconstruction,
+#     PtychographyOptimizer,
+#     SimultaneousPtychographicReconstruction,
+#     SingleslicePtychographicReconstruction,
+# )
 
 # polar
 from py4DSTEM.process.polar import PolarDatacube
@@ -100,7 +101,8 @@
 from py4DSTEM.process.strain.strain import StrainMap
 
 # TODO WPF
-# from py4DSTEM.process.wholepatternfit import (
+from py4DSTEM.process import wholepatternfit
+#  import (
 #     ComplexOverlapKernelDiskLattice,
 #     DCBackground,
 #     Flag,