Merge branch 'Chocologism-feature/psuedo_isothermal_test'

Author: Jammy2211

autogalaxy/operate/image.py

@@ -186,21 +186,14 @@ def visibilities_from(
             in the uv-plane.
         """
 
-        image_2d = self.image_2d_from(grid=grid)
+        from autogalaxy.profiles.light.abstract import LightProfile
 
-        if jnp.any(image_2d.array):
-            return transformer.visibilities_from(image=image_2d)
-        return aa.Visibilities.zeros(shape_slim=(transformer.uv_wavelengths.shape[0],))
+        if self.has(cls=LightProfile):
 
-        # return jax.lax.cond(
-        #     jnp.any(image_2d.array),
-        #     lambda _: transformer.visibilities_from(image=image_2d),
-        #     lambda _: aa.Visibilities.zeros(
-        #         shape_slim=(transformer.uv_wavelengths.shape[0],)
-        #     ),
-        #     operand=None,
-        # )
+            image_2d = self.image_2d_from(grid=grid)
+            return transformer.visibilities_from(image=image_2d)
 
+        return aa.Visibilities.zeros(shape_slim=(transformer.uv_wavelengths.shape[0],))
 
 class OperateImageList(OperateImage):
     """

autogalaxy/profiles/mass/__init__.py

@@ -1,8 +1,11 @@
 from .abstract.abstract import MassProfile
 from .point import PointMass, SMBH, SMBHBinary
 from .total import (
-    dPIE,
-    dPIESph,
+    dPIEMD,
+    dPIEMDSph,
+    PIEMD,
+    dPIEP,
+    dPIEPSph,
     PowerLawCore,
     PowerLawCoreSph,
     PowerLawBroken,

autogalaxy/profiles/mass/total/__init__.py

@@ -1,4 +1,5 @@
 from .dual_pseudo_isothermal import dPIE, dPIESph
+from .pseudo_isothermal import PIEMD, dPIEMD, dPIEMDSph, dPIEP, dPIEPSph
 from .isothermal import Isothermal, IsothermalSph
 from .isothermal_core import IsothermalCore, IsothermalCoreSph
 from .power_law import PowerLaw, PowerLawSph