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