wintertoo 1.3.2__py3-none-any.whl → 1.5.0__py3-none-any.whl

wintertoo/data/__init__.py

@@ -32,7 +32,7 @@ WINTER_BASE_WIDTH = 1.0
 
 MAX_TARGNAME_LEN = 30
 
-WinterImageTypes = Literal["exposure", "raw", "science", "stack", "diff"]
+WinterImageTypes = Literal["exposure", "raw", "science", "stack", "diff", "avro"]
 DEFAULT_IMAGE_TYPE = "stack"
 
 PROGRAM_DB_HOST = "jagati.caltech.edu"

wintertoo/data/observing_request_schema.json

@@ -5,7 +5,7 @@
     "raDeg": {"type": "number", "comment": "Degrees (decimal)"},
     "decDeg": {"type": "number", "comment": "Degrees (decimal)"},
     "filter": {"type": "string", "comment": "'u'/'g'/'r'/'i'/'Y'/'J'/'H'/'dark'"},
-    "visitExpTime": {"type": "number", "comment": "Total observation time (seconds)", "default": 30.0},
+    "visitExpTime": {"type": "number", "comment": "Total observation time (seconds)", "default": 960.0},
     "priority": {"type": "number", "comment": "Priority of observation"},
     "progPI": {"type": "string", "comment": "PI of observation"},
     "progID": {"type": "integer", "comment": "Between 0 and 4 (0-Calibration, 1-Survey, 2-MIT, 3-Caltech, 4-Engineering)"},
@@ -14,7 +14,7 @@
     "validStop": {"type": "number", "comment": "Format: MJD"},
     "observed": {"anyOf": [{"type":  "boolean", "default":  false}, {"type" : "integer", "minimum" : 0, "maximum" : 1}]},
     "maxAirmass": {"type":  "number", "comment": "Maximum airmass for observation", "default":  2.0},
-    "ditherNumber": {"type": "integer", "default": 1},
+    "ditherNumber": {"type": "integer", "default": 8},
     "ditherStepSize": {"type": "number", "comment": "arcsec", "default": 30.0},
     "fieldID": {"type": "integer", "default": 999999999},
     "targName": {"type": ["string", "null"], "comment": "Target name e.g. GW170817", "default": null},

wintertoo/models/image.py

@@ -6,7 +6,7 @@ from typing import Optional
 
 from astropy import units as u
 from astropy.time import Time
-from pydantic import BaseModel, Field, model_validator
+from pydantic import AliasChoices, BaseModel, Field, model_validator
 
 from wintertoo.data import DEFAULT_IMAGE_TYPE, MAX_TARGNAME_LEN, WinterImageTypes
 from wintertoo.errors import WinterValidationError
@@ -39,8 +39,10 @@ class ProgramImageQuery(BaseModel):
         default=get_date(Time.now()),
         examples=[get_date(Time.now() - 30.0 * u.day), get_date(Time.now())],
     )
-    kind: WinterImageTypes = Field(
-        default=DEFAULT_IMAGE_TYPE, example="raw/science/diff"
+    image_type: WinterImageTypes = Field(
+        default=DEFAULT_IMAGE_TYPE,
+        example="raw/science/diff",
+        validation_alias=AliasChoices("image_type", "kind"),
     )
 
     @model_validator(mode="after")

wintertoo/models/too.py

@@ -5,7 +5,14 @@ Models for ToO requests
 from typing import List, Optional, Union
 
 from astropy.time import Time
-from pydantic import BaseModel, ConfigDict, Field, model_validator
+from pydantic import (
+    AliasChoices,
+    BaseModel,
+    ConfigDict,
+    Field,
+    computed_field,
+    model_validator,
+)
 
 from wintertoo.data import (
     MAX_TARGNAME_LEN,
@@ -39,15 +46,21 @@ class ToORequest(BaseModel):
         examples=["SN2021abc", "ZTF19aapreis"],
         default=get_default_value("targName"),
     )
-    t_exp: float = Field(
+    total_exposure_time: float = Field(
         default=get_default_value("visitExpTime"),
         title="Combined Exposure Time across dithers (s)",
         ge=1.0,
+        validation_alias=AliasChoices("total_exposure_time", "t_exp"),
     )
-    n_exp: int = Field(default=1, ge=1, title="Number of dither sets")
     n_dither: int = Field(
         default=get_default_value("ditherNumber"), ge=1, title="Number of dithers"
     )
+    n_repetitions: int = Field(
+        default=1,
+        ge=1,
+        title="Number of repeated dither sets",
+        validation_alias=AliasChoices("n_repetitions", "n_exp"),
+    )
     dither_distance: float = Field(
         get_default_value("ditherStepSize"), ge=0.0, title="dither distance (arcsec)"
     )
@@ -69,6 +82,17 @@ class ToORequest(BaseModel):
         title="Place ra/dec at the center of the best detector",
     )
 
+    @computed_field
+    @property
+    def single_exposure_time(self) -> float:
+        """
+        Computed field to get the exposure time per dither.
+        Equivalent to total_exposure_time/n_dither
+
+        :return: Exposure time per dither
+        """
+        return self.total_exposure_time / self.n_dither
+
     @model_validator(mode="after")
     def validate_end_time(self):
         """
@@ -93,7 +117,7 @@ class ToORequest(BaseModel):
         :return: Validated total exposure time per dither set
         """
         n_dithers = self.n_dither
-        t_exp = self.t_exp
+        t_exp = self.total_exposure_time
         t_per_dither = t_exp / n_dithers
 
         if t_per_dither > MAX_EXPOSURE_TIME:

wintertoo/schedule.py

@@ -43,14 +43,15 @@ def make_schedule(
 
     for too in toos:
         for filter_name in too.filters:
-            for _ in range(too.n_exp):
+            for _ in range(too.n_repetitions):
                 new = {
                     "targName": too.target_name,
                     "raDeg": too.ra_deg,
                     "decDeg": too.dec_deg,
                     "fieldID": too.field_id,
                     "filter": filter_name,
-                    "visitExpTime": too.t_exp,
+                    "visitExpTime": too.total_exposure_time,
+                    "singleExpTime": too.single_exposure_time,
                     "priority": too.target_priority,
                     "progPI": program.pi_name,
                     "progName": program.progname,
@@ -121,7 +122,10 @@ def schedule_ra_dec(
     else:
         field_id = get_default_value("fieldID")
 
-    full_request = FullTooRequest(field_id=field_id, **too.model_dump())
+    full_request = FullTooRequest(
+        field_id=field_id,
+        **too.model_dump(exclude=set(too.model_computed_fields.keys())),
+    )
 
     schedule = make_schedule(
         toos=[full_request],
@@ -152,7 +156,11 @@ def schedule_field(
     ra_deg = float(field_details["RA"].iloc[0])
     dec_deg = float(field_details["Dec"].iloc[0])
 
-    full_request = FullTooRequest(ra_deg=ra_deg, dec_deg=dec_deg, **too.model_dump())
+    full_request = FullTooRequest(
+        ra_deg=ra_deg,
+        dec_deg=dec_deg,
+        **too.model_dump(exclude=set(too.model_computed_fields.keys())),
+    )
 
     schedule = make_schedule(
         toos=[full_request],

{wintertoo-1.3.2.dist-info → wintertoo-1.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: wintertoo
-Version: 1.3.2
+Version: 1.5.0
 Author-email: Robert Stein <rdstein@caltech.edu>, Danielle Frostig <frostig@mit.edu>, Viraj Karambelkar <viraj@astro.caltech.edu>
 License: MIT
 Project-URL: homepage, https://github.com/winter-telescope/wintertoo

{wintertoo-1.3.2.dist-info → wintertoo-1.5.0.dist-info}/RECORD

@@ -2,20 +2,20 @@ wintertoo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wintertoo/database.py,sha256=Ng8qsnDN3p4q-hXRgsHvElzr3k_8H14vq3-C2XChkr8,3020
 wintertoo/errors.py,sha256=iAFcy0z3A3bCnkJmAt_P-77fPaPHFGS1JbrK6O2dCEM,223
 wintertoo/fields.py,sha256=TDSOwJeR4UvfzKQCaIqoPLv0hSU_73TZwQcZ2FbnO8Q,6029
-wintertoo/schedule.py,sha256=Ll_8udFyn14uq3cVri9tVbsz6svluhkX2XXs8JFtLUU,5584
+wintertoo/schedule.py,sha256=0nqOh_Ih4Ce0UMVXIzU2SfhdibFfKciPqQhEiAG9iyU,5813
 wintertoo/submit.py,sha256=kd6_aziOf5QcemsyyhSRSnbXlLj8OWsY1_Gj0ZXsUuo,3201
 wintertoo/utils.py,sha256=BWgBZoSX5xfqURb2Nkvf3i6A_IcVYU2SoXFvp6ez7Xc,3306
 wintertoo/validate.py,sha256=YpSABluVXxl33j_ht7TwfC-0SlN4IC0BnbbQiTwUk4c,10476
-wintertoo/data/__init__.py,sha256=7998yomm2Kb0_687D82tjIhqwO-LOh7pru1sNUC1wI0,1594
-wintertoo/data/observing_request_schema.json,sha256=R99sh_XEWVcUEP7eSGtFBw-SMa8M1CQRKfBESdyL9WU,1732
+wintertoo/data/__init__.py,sha256=OPsHalwB9mTMBHp-W-iAbsP39hrgQ4myNt1Daxkwz2s,1602
+wintertoo/data/observing_request_schema.json,sha256=00laKF0S5O6lolsUrimReR67gymCEJuH4z3u6PUFgFA,1733
 wintertoo/data/summer_fields.txt,sha256=5Sc7MBUacelzaq1KHSLmfAyZ3WB5YifMNwKRjBRBcRk,52684603
 wintertoo/data/winter_fields.txt,sha256=TxySQTmJXCCgaf-oC1gzOYQb2Vr26KEKdJxqrZHHet0,3529364
 wintertoo/models/__init__.py,sha256=c2GEbIHvOT34IhpiMyqA7Tn83hEmnwx6Q7BuxURp97k,379
-wintertoo/models/image.py,sha256=YNX9a5nwAjWw2disbyNbrDsJUhmMwIuBedTGScCzlD4,3836
+wintertoo/models/image.py,sha256=KG07tysYHPiFcQplkC4aZ9NjoQLb8TNl2Cy5WFU2a5w,3926
 wintertoo/models/program.py,sha256=mtewVnt7NxSK0hxX8aUO8qehEtOi_Cieli4voCCRh4I,1804
-wintertoo/models/too.py,sha256=H9HctFxXL4UPqgjs7Xr8oW0iVLJ8BbLv9oza8HLzBW4,5513
-wintertoo-1.3.2.dist-info/LICENSE,sha256=1-4yY2S7St-8Koy4JGheNG9QPkwff7831u2JITQ2oQs,1063
-wintertoo-1.3.2.dist-info/METADATA,sha256=Lli0aWNu7SD4lcILhfCzJw87SMQBijKwPRkMRbPIjVQ,2498
-wintertoo-1.3.2.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-wintertoo-1.3.2.dist-info/top_level.txt,sha256=6SMjMBzaNrD77erdCiVcRTrPZ-x98SDX43akRjWj4T8,10
-wintertoo-1.3.2.dist-info/RECORD,,
+wintertoo/models/too.py,sha256=SXaAx_nkPyqYaUS7UChA86TkeKrIOqvxccrNXaXXvbo,6098
+wintertoo-1.5.0.dist-info/LICENSE,sha256=1-4yY2S7St-8Koy4JGheNG9QPkwff7831u2JITQ2oQs,1063
+wintertoo-1.5.0.dist-info/METADATA,sha256=rgU0vbaECPDD6ye9eVTVGCwPKSa9BGvBe1c4rcRCiKg,2498
+wintertoo-1.5.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+wintertoo-1.5.0.dist-info/top_level.txt,sha256=6SMjMBzaNrD77erdCiVcRTrPZ-x98SDX43akRjWj4T8,10
+wintertoo-1.5.0.dist-info/RECORD,,