Add the required import statements

CDTools/models/__init__.py

@@ -27,6 +27,7 @@ defining a new ptychography model before attempting to do so.
 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.bragg_2d_ptycho import Bragg2DPtycho
 from CDTools.models.multislice_2d_ptycho import Multislice2DPtycho
 from CDTools.models.rpi import RPI

CDTools/models/polarized_fancy_ptycho.py

@@ -1,6 +1,3 @@
-cd ..
-_future__ import division, print_function, absolute_import
-
 import torch as t
 from CDTools.models import CDIModel, FancyPtycho
 from CDTools.datasets import Ptycho2DDataset

CDTools/tools/__init__.py

@@ -14,9 +14,6 @@ having numpy and torch defined under CDTools.tools.cmath, you know?
 
 """
 
-from __future__ import division, print_function, absolute_import
-
-
 from CDTools.tools import losses
 from CDTools.tools import data
 from CDTools.tools import image_processing
@@ -27,3 +24,4 @@ from CDTools.tools import propagators
 from CDTools.tools import measurements
 from CDTools.tools import analysis
 from CDTools.tools import atoms
+from CDTools.tools import polarization

CDTools/tools/image_processing/__init__.py

@@ -1,3 +1 @@
-from __future__ import division, print_function, absolute_import
-
 from CDTools.tools.image_processing.image_processing import *

CDTools/tools/polarization/__init__.py

@@ -0,0 +1 @@
+from CDTools.tools.polarization.polarization import *