dagster-shared 1.12.5__py3-none-any.whl → 1.12.6__py3-none-any.whl

dagster_shared/dagster_model/pydantic_compat_layer.py

@@ -60,6 +60,10 @@ class ModelFieldCompat:
         if hasattr(self.field, "discriminator"):
             return self.field.discriminator if hasattr(self.field, "discriminator") else None
 
+    @property
+    def json_schema_extra(self) -> Optional[dict[str, Any]]:
+        return getattr(self.field, "json_schema_extra", None)
+
 
 def model_fields(model: type[BaseModel]) -> dict[str, ModelFieldCompat]:
     """Returns a dictionary of fields for a given pydantic model, wrapped

dagster_shared/record/__init__.py

@@ -32,7 +32,7 @@ _NAMED_TUPLE_BASE_NEW_FIELD = "__nt_new__"
 _REMAPPING_FIELD = "__field_remap__"
 _ORIGINAL_CLASS_FIELD = "__original_class__"
 _KW_ONLY_FIELD = "__kw_only__"
-
+_REPLACE_NEW_FIELD = "__replace_new__"
 
 _sample_nt = namedtuple("_canary", "x")
 # use a sample to avoid direct private imports (_collections._tuplegetter)
@@ -173,6 +173,7 @@ def _namedtuple_record_transform(
         _REMAPPING_FIELD: field_to_new_mapping or {},
         _ORIGINAL_CLASS_FIELD: cls,
         _KW_ONLY_FIELD: kw_only,
+        _REPLACE_NEW_FIELD: generated_new if generated_new else nt_new,
         "__reduce__": _reduce,
         # functools doesn't work, so manually update_wrapper
         "__module__": cls.__module__,
@@ -489,13 +490,8 @@ def replace(obj: TVal, **kwargs) -> TVal:
     cls = obj.__class__
 
     # if we have runtime type checking, go through that to vet new field values
-    if hasattr(cls, _CHECKED_NEW):
-        target = _CHECKED_NEW
-    else:
-        target = _NAMED_TUPLE_BASE_NEW_FIELD
-
-    return getattr(cls, target)(
-        obj.__class__,
+    return getattr(cls, _REPLACE_NEW_FIELD)(
+        cls,
         **{**as_dict(obj), **kwargs},
     )
 
@@ -570,6 +566,7 @@ class JitCheckedNew:
         for c in cls.__mro__:
             if c.__new__ is self:
                 c.__new__ = compiled_fn
+        setattr(cls, _REPLACE_NEW_FIELD, compiled_fn)
 
         return compiled_fn(cls, *args, **kwargs)
 

dagster_shared/version.py

@@ -1 +1 @@
-__version__ = "1.12.5"
+__version__ = "1.12.6"

{dagster_shared-1.12.5.dist-info → dagster_shared-1.12.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dagster_shared
-Version: 1.12.5
+Version: 1.12.6
 Summary: Shared code between dagster and dagster-dg-core.
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-shared
 Author: Dagster Labs

{dagster_shared-1.12.5.dist-info → dagster_shared-1.12.6.dist-info}/RECORD

@@ -5,7 +5,7 @@ dagster_shared/match.py,sha256=P9wQSvzb7a03jXlC-0w9msoX5bV_BE1uSvdYvavNO6M,3444
 dagster_shared/merger.py,sha256=c9cJTwHtgMXgR9MtB-KwZEMtQSXn13rF08F2ID5aDTg,1758
 dagster_shared/modules.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dagster_shared/py.typed,sha256=mDShSrm8qg9qjacQc2F-rI8ATllqP6EdgHuEYxuCXZ0,7
-dagster_shared/version.py,sha256=4v8TGQDepA1SecAP1bl6pAHVU5O905enxAg_EyVjJZw,23
+dagster_shared/version.py,sha256=K-O_Zmi1DuK_mEBcDOT_b5zNp9ONlW-73xBIaMAX9Fo,23
 dagster_shared/check/README.md,sha256=UlxRVMWJEoVAlaHWDvhobUpuNQGBG5-ddqZ6OIfLYnM,1352
 dagster_shared/check/__init__.py,sha256=CHSOqJpaeWr6skJT7jD8HnN5s4bdnoX4x94sGN55KqA,3482
 dagster_shared/check/builder.py,sha256=2Ne2EM_2tGm2zYMg0vRxqC5r-KLc746UNnXe1Fk_vRc,17311
@@ -14,13 +14,13 @@ dagster_shared/check/functions.py,sha256=PhFZkMag2q5MKulQjgyIs_O9_t0XqOnzNRdQFZT
 dagster_shared/check/record.py,sha256=vzwbmxQNQl3_oG74NRKgjlIDM2xpL5ZbKtKhoFNhNq4,412
 dagster_shared/cli/__init__.py,sha256=y2q8V4gn5IRxq962uGjMkgWqpMK0B3pongr2mMB2vvg,8781
 dagster_shared/dagster_model/__init__.py,sha256=MoUreIsyzBIjn1w2noiyR0CmBg2kiP2VhUGkXvFI5E8,1296
-dagster_shared/dagster_model/pydantic_compat_layer.py,sha256=9c2BmeucYvvfnmeSffOqadKX0Kitx1rBu-WW32A116Y,2892
+dagster_shared/dagster_model/pydantic_compat_layer.py,sha256=nA3s7rOStkjNAOGNBlS3VhukBawRt1FAXkgLarDtxqY,3030
 dagster_shared/libraries/__init__.py,sha256=VBOVSB2MU2HZzUTarK-3JE-7k7xkhiggPZnaAsEcwNg,4669
 dagster_shared/plus/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dagster_shared/plus/config.py,sha256=0kAWocZFYh9NmCoOT348YUg0vQe9SEpJ_Ze9R3uyR-w,5756
 dagster_shared/plus/config_utils.py,sha256=eVvahsrGQdcYL1H3_63kXO2NKsVPtWXhlGFajvJPQ5w,6147
 dagster_shared/plus/login_server.py,sha256=zRc85NUHtxpgXGfzV1d4hq-n9ZdXTBiSmagUebwxqdE,3630
-dagster_shared/record/__init__.py,sha256=bOQHMcYGHKe5d3tasHNIPfXJAJUPvb7B1puZPjzdlFA,22576
+dagster_shared/record/__init__.py,sha256=yfmDqHKt1fgwAzz9ACKicF61eRqcYqBLGY49F6JtliQ,22621
 dagster_shared/scaffold/__init__.py,sha256=xb5-ZdbZ5wSYYJy8VA1KGDnPodM6bKTCSBPS85Wqi_w,3759
 dagster_shared/serdes/__init__.py,sha256=BInlrdN_14uwPz0t32Dl1VH4D-7_rG1vUfZ0rw5DQc0,649
 dagster_shared/serdes/errors.py,sha256=qg_VZqtuYAlAXIjOp96bGs_z309ifDoHuT4sVnxJ6nA,142
@@ -47,8 +47,8 @@ dagster_shared/utils/warnings.py,sha256=yxiG8ghc3RwyomRItwIKQe6TrUdTlJAUX3BssOwe
 dagster_shared/yaml_utils/__init__.py,sha256=Ya7qT3aJGPC1TgD-pSkkhLX5U4QbrUkCsrUtZQeACXg,10529
 dagster_shared/yaml_utils/sample_yaml.py,sha256=YC-1h1P48F3QV-Uwo-MqzEkzlrMliXPfZM_X6ynZvTQ,5461
 dagster_shared/yaml_utils/source_position.py,sha256=yVSnHUkBYzWzLzqtyGbqRjH2b7oTeiE4K3b-uokYUIQ,9786
-dagster_shared-1.12.5.dist-info/licenses/LICENSE,sha256=hgU51ohULAFKZE2la8sartPYoUaAA14jxFAfFC4Tow0,11347
-dagster_shared-1.12.5.dist-info/METADATA,sha256=NWO8Qvp5Nwj6N58b5p6TKsZzaerPfhw9TAkmTybDD78,1153
-dagster_shared-1.12.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-dagster_shared-1.12.5.dist-info/top_level.txt,sha256=S8ADcO4aTW07ONhLHXDxiuSkDzucgYKOlocB2x3r_dY,15
-dagster_shared-1.12.5.dist-info/RECORD,,
+dagster_shared-1.12.6.dist-info/licenses/LICENSE,sha256=hgU51ohULAFKZE2la8sartPYoUaAA14jxFAfFC4Tow0,11347
+dagster_shared-1.12.6.dist-info/METADATA,sha256=DLej44NBUiK4RjPyDZWiAVGr6MF7f4YmLomcZXdKeMg,1153
+dagster_shared-1.12.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+dagster_shared-1.12.6.dist-info/top_level.txt,sha256=S8ADcO4aTW07ONhLHXDxiuSkDzucgYKOlocB2x3r_dY,15
+dagster_shared-1.12.6.dist-info/RECORD,,