Factory for (SourceGalaxy)SelectField

firecrown/likelihood/number_counts.py

@@ -22,6 +22,7 @@
     SourceGalaxySystematic,
     PhotoZShiftFactory,
     PhotoZShiftandStretchFactory,
+    SourceGalaxySelectFieldFactory,
     Tracer,
 )
 from firecrown.metadata_types import InferredGalaxyZDist, TypeSource
@@ -613,7 +614,8 @@ def create_global(self) -> ConstantMagnificationBiasSystematic:
     | LinearBiasSystematicFactory
     | PTNonLinearBiasSystematicFactory
     | MagnificationBiasSystematicFactory
-    | ConstantMagnificationBiasSystematicFactory,
+    | ConstantMagnificationBiasSystematicFactory
+    | SourceGalaxySelectFieldFactory,
     Field(discriminator="type", union_mode="left_to_right"),
 ]
 

firecrown/likelihood/source.py

@@ -476,6 +476,27 @@
         return replace(tracer_arg, field=self.field)
 
 
+class SourceGalaxySelectFieldFactory(BaseModel):
+    """Factory class for SourceGalaxySelectField objects."""
+
+    model_config = ConfigDict(extra="forbid", frozen=True)
+
+    field_name: str
+
+    type: Annotated[
+        Literal["SourceGalaxySelectFieldFactory"],
+        Field(description="The type of the systematic."),
+    ] = "SourceGalaxySelectFieldFactory"
+
+    def create(self, bin_name: str) -> SourceGalaxySelectField:
+        """Create a SourceGalaxySelectField object with the given field name."""
+        raise ValueError("SourceGalaxySelectField cannot be per-bin.")
+
+    def create_global(self) -> SourceGalaxySelectField:
+        """Create a PhotoZShift object with the given field name."""
+        return SourceGalaxySelectField(field=self.field_name)
+
+
 class SourceGalaxy(Source, Generic[_SourceGalaxyArgsT]):
     """Source class for galaxy based sources."""
 

firecrown/likelihood/weak_lensing.py

@@ -24,6 +24,7 @@
     SourceGalaxySystematic,
     PhotoZShiftFactory,
     PhotoZShiftandStretchFactory,
+    SourceGalaxySelectFieldFactory,
     Tracer,
 )
 from firecrown.metadata_types import InferredGalaxyZDist, TypeSource
@@ -570,7 +571,8 @@ def create_global(self) -> TattAlignmentSystematic:
     | PhotoZShiftandStretchFactory
     | MultiplicativeShearBiasFactory
     | LinearAlignmentSystematicFactory
-    | TattAlignmentSystematicFactory,
+    | TattAlignmentSystematicFactory
+    | SourceGalaxySelectFieldFactory,
     Field(discriminator="type", union_mode="left_to_right"),
 ]