Set default polarization to unpolarized

Author: PreisTo

astromodels/core/model_parser.py

@@ -723,7 +723,7 @@ def _parse_polarization(self, polarization_definititon):
 
             # just make a default polarization
 
-            this_polarization = polarization.Polarization()
+            this_polarization = polarization.Unpolarized()
             # raise ModelSyntaxError("Polarization specification for source %s has an
             # invalid parameters. You need to specify either 'angle' and 'degree', or
             # 'I' ,'Q', 'U' and 'V'." % self._source_name)
@@ -777,7 +777,7 @@ def _parse_spectral_component(self, component_name, component_definition):
 
         else:
 
-            this_polarization = polarization.Polarization()
+            this_polarization = polarization.Unpolarized()
 
         this_spectral_component = spectral_component.SpectralComponent(
             component_name, shape, this_polarization

astromodels/core/polarization.py

@@ -8,12 +8,13 @@
 
 
 class Polarization(Node):
-    def __init__(self, polarization_type="linear"):
+    def __init__(self, polarization_type="unpolarized"):
 
         assert polarization_type in [
+            "unpolarized",
             "linear",
             "stokes",
-        ], "polarization must be linear or stokes"
+        ], "polarization must be unpolarized, linear or stokes"
 
         self._polarization_type = polarization_type
 

astromodels/core/spectral_component.py

@@ -1,6 +1,6 @@
 __author__ = "giacomov"
 
-from astromodels.core.polarization import Polarization
+from astromodels.core.polarization import Unpolarized
 from astromodels.core.tree import Node
 
 
@@ -25,7 +25,7 @@ def __init__(self, name, shape, polarization=None):
 
         if polarization is None:
 
-            self._polarization = Polarization()
+            self._polarization = Unpolarized()
 
         else: