dkist-processing-common 10.8.4rc1__py3-none-any.whl → 11.0.0rc1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (29) hide show
  1. changelog/256.feature.rst +2 -0
  2. changelog/257.feature.rst +1 -0
  3. changelog/259.feature.rst +1 -0
  4. dkist_processing_common/models/dkist_location.py +27 -0
  5. dkist_processing_common/models/metric_code.py +1 -0
  6. dkist_processing_common/models/quality.py +49 -5
  7. dkist_processing_common/parsers/dsps_repeat.py +1 -1
  8. dkist_processing_common/parsers/id_bud.py +1 -1
  9. dkist_processing_common/parsers/near_bud.py +8 -6
  10. dkist_processing_common/parsers/retarder.py +1 -1
  11. dkist_processing_common/parsers/time.py +26 -9
  12. dkist_processing_common/parsers/unique_bud.py +8 -6
  13. dkist_processing_common/parsers/wavelength.py +1 -1
  14. dkist_processing_common/tasks/mixin/quality/_base.py +7 -1
  15. dkist_processing_common/tasks/mixin/quality/_metrics.py +129 -0
  16. dkist_processing_common/tasks/parse_l0_input_data.py +4 -3
  17. dkist_processing_common/tasks/write_l1.py +6 -35
  18. dkist_processing_common/tests/test_assemble_quality.py +33 -4
  19. dkist_processing_common/tests/test_dkist_location.py +15 -0
  20. dkist_processing_common/tests/test_quality_mixin.py +204 -1
  21. dkist_processing_common/tests/test_stems.py +141 -13
  22. dkist_processing_common/tests/test_write_l1.py +0 -12
  23. {dkist_processing_common-10.8.4rc1.dist-info → dkist_processing_common-11.0.0rc1.dist-info}/METADATA +3 -2
  24. {dkist_processing_common-10.8.4rc1.dist-info → dkist_processing_common-11.0.0rc1.dist-info}/RECORD +26 -24
  25. changelog/235.feature.rst +0 -3
  26. changelog/235.misc.1.rst +0 -2
  27. changelog/235.misc.rst +0 -1
  28. {dkist_processing_common-10.8.4rc1.dist-info → dkist_processing_common-11.0.0rc1.dist-info}/WHEEL +0 -0
  29. {dkist_processing_common-10.8.4rc1.dist-info → dkist_processing_common-11.0.0rc1.dist-info}/top_level.txt +0 -0
@@ -1,7 +1,7 @@
1
1
  changelog/.gitempty,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
- changelog/235.feature.rst,sha256=F27MUnNkqjcmlmRQ6NQUi2wd5M9dOjN2jZ70TV95Y20,365
3
- changelog/235.misc.1.rst,sha256=kQujVYRa_axHQSRFxW3SNPNMgEL-YgtnyLud-aJljL8,234
4
- changelog/235.misc.rst,sha256=ooK2tbA5EEFyYNZLLvgYiMZ-Eza5UzPHN9MvpuX-cu4,92
2
+ changelog/256.feature.rst,sha256=i_Nf4STs2fP2MNZ3Ec4I5K4Avh0oLD8LF4i8G_x19VM,175
3
+ changelog/257.feature.rst,sha256=5RIxMuwnHi6lrTZ_RC1GVRqx9nFHM--ShQEfdaVGqv0,90
4
+ changelog/259.feature.rst,sha256=G2UIyD1cpP8ptpDk39bSI4TvqLt46HAyjWJIQ7VSVXs,95
5
5
  dkist_processing_common/__init__.py,sha256=490Fwm_GgqpwriQlsYfKcLUZNhZ6GkINtJqcYSIEKoU,319
6
6
  dkist_processing_common/config.py,sha256=IcpaD_NvHZU-aLlUNOTdRC4V7ADIvVQwrZ2dHhIr4NY,4247
7
7
  dkist_processing_common/manual.py,sha256=M7FW1viESaTfK1jLqHLp7JMGTGeoTxHtgCXRjZpDR8g,6990
@@ -25,6 +25,7 @@ dkist_processing_common/fonts/Lato-Regular.ttf,sha256=1jbkaDIx-THtoiLViOlE0IK_07
25
25
  dkist_processing_common/fonts/__init__.py,sha256=hBvZRtkoGRPlNDWCK-ZePXdSIlThCcjwBDfYaamVgAw,101
26
26
  dkist_processing_common/models/__init__.py,sha256=6LMqemdzVZ87fRrpAsbEnTtWZ02_Gu_oajsUlwGRH_Q,74
27
27
  dkist_processing_common/models/constants.py,sha256=1Eb8RDeuCr6brl237iGKxYLWCH49I6bOUEj_Tv-zFbQ,5441
28
+ dkist_processing_common/models/dkist_location.py,sha256=3cbN6As60tJa0uoOSNFNnz2sqSyZvPW-vDMCBrQIZW0,971
28
29
  dkist_processing_common/models/fits_access.py,sha256=Au9JROwhVla9zb_u0dN8mIWiSJd_Pca0oOr4N1hN0HY,4113
29
30
  dkist_processing_common/models/flower_pot.py,sha256=59C5uGYKyMyncqQYxhzDZWl8k1DRZFB6s9RF-HFp9mY,5128
30
31
  dkist_processing_common/models/fried_parameter.py,sha256=ro_H2Eo3I88lRf1wJjZfTc_XOjhgLt4whIQR_sjAFbM,1609
@@ -32,56 +33,57 @@ dkist_processing_common/models/graphql.py,sha256=BBJBIBADAPQAskqS8Qh58DYEyFjY9GY
32
33
  dkist_processing_common/models/input_dataset.py,sha256=OZDxyjHZfFrksFGpas1gsB14Q77CeNsk_nI-EYo3ZRI,4121
33
34
  dkist_processing_common/models/message.py,sha256=DRW7Qhl01dF5KagcqLta5U-uzdOMewrsHvMatDT6jnk,1684
34
35
  dkist_processing_common/models/message_queue_binding.py,sha256=ROQ2ZQE3TCr4gVbz4WggvUSExAiWP8SD_GjjQl482M8,1012
35
- dkist_processing_common/models/metric_code.py,sha256=McXAEF1Sa0_YlR1niXYLJWLFHhdLQhmYw9Xtpr5FGws,815
36
+ dkist_processing_common/models/metric_code.py,sha256=mF-i0nuzko7K949aS01hEsssGes_PIF5B0b_mVqwoL4,847
36
37
  dkist_processing_common/models/parameters.py,sha256=dTup7mPTEmySXP0O3j4sPPY8Jqe2zf-3sr2hjeMatEY,9649
37
- dkist_processing_common/models/quality.py,sha256=ONz1A6_qyEoZhQkVp9LChAgm93aGt1O5WSRneE3XCCA,2319
38
+ dkist_processing_common/models/quality.py,sha256=wI779KvakvZNOJXq8qgdsF8u9sVYAzd8ezhoXpiccqQ,3994
38
39
  dkist_processing_common/models/tags.py,sha256=ykOYqWMU7_ffvRCv84-avjXyty9pHBo7EXwsjIjStjs,12058
39
40
  dkist_processing_common/models/task_name.py,sha256=NL0n92A9vVYBV-yvh8d-qFOCxVy0X2GECDmLgIzrmOY,565
40
41
  dkist_processing_common/models/wavelength.py,sha256=Wtmu5QhjPpsqIGfUQ0Wh-3PQlGeRdGV9BfFAy23HLGg,966
41
42
  dkist_processing_common/parsers/__init__.py,sha256=XJQzHtPb78F6-qXXKXjyztc0x-aHVlgv1C_l4dR88tI,67
42
43
  dkist_processing_common/parsers/cs_step.py,sha256=en1ovwy8H2jeUxZd0XDSV8Qv3ZawRm03q6wpJj4a3C8,6461
43
- dkist_processing_common/parsers/dsps_repeat.py,sha256=NiUMnfMYSn0qserHM735V1Z6BCfw4CAanlqtjfyRkos,1571
44
+ dkist_processing_common/parsers/dsps_repeat.py,sha256=fcOya_70VuzShe_LL6iUpJ5vysAeHXDIg8wuC2fbeHk,1572
44
45
  dkist_processing_common/parsers/experiment_id_bud.py,sha256=IpDhqcHxriYn8fVFDQLgUiEiX98TLafAMiLoxuvELjo,706
45
- dkist_processing_common/parsers/id_bud.py,sha256=cmq44quAGHZJ85hgQfpgh3NrSsrf_xVJyE8j6e3g_og,1536
46
+ dkist_processing_common/parsers/id_bud.py,sha256=nmfpejWWhELK3uHnBmuExxKyo_x8IJxkY7RTK_Ptcm0,1537
46
47
  dkist_processing_common/parsers/l0_fits_access.py,sha256=mhgezAA0X5zvgEmZFPIgKYpngw3l4pIKcXIr_X7J8R0,921
47
48
  dkist_processing_common/parsers/l1_fits_access.py,sha256=o0_7XfcCcuKpuN8K0E_dfEnPvB1a9VxqKBaxpRGDOL0,2595
48
- dkist_processing_common/parsers/near_bud.py,sha256=gLTr7EILVCREAhmE0VaU7-KKc7aosoMmZmQtJEXxs2I,3733
49
+ dkist_processing_common/parsers/near_bud.py,sha256=aiyg19SNhFATvAyd0yLozisXVwQfnX86k1lWxUaInRw,3865
49
50
  dkist_processing_common/parsers/proposal_id_bud.py,sha256=61uRxT538UDiEg3_By_asSzQ96xkAuCEEZZTrk0UF0E,677
50
51
  dkist_processing_common/parsers/quality.py,sha256=Lg5z53kAvJuw-lPrX9HSkkFi0Bk0D5oDv8TAIgbLOMo,1140
51
- dkist_processing_common/parsers/retarder.py,sha256=OzhdC8VoV2zdcvq8Nn5Ghc-Q3BFfMRITLx5jMHrWNqI,1440
52
+ dkist_processing_common/parsers/retarder.py,sha256=D1EV2e4Yomww1WTdq2aWaQ_2FYUf7fRna1juM48VU58,1441
52
53
  dkist_processing_common/parsers/single_value_single_key_flower.py,sha256=cQX_C_Ms6EIOCW5TTDo6PnLoJ6In6MQTShKdMvt57OU,1230
53
54
  dkist_processing_common/parsers/task.py,sha256=RTNGSg3BUpEYLOcolN79Wc16s86FqRcDP-H7Hj9ovGg,3938
54
- dkist_processing_common/parsers/time.py,sha256=9MQWyBeInBaqZjYxdWbLTOvxol0-q35VvIAyGL28CIs,7839
55
- dkist_processing_common/parsers/unique_bud.py,sha256=QFsBEdCyx2GzYOUVobFxSxlKyaHT1F8i5PQegmwMlhM,3041
56
- dkist_processing_common/parsers/wavelength.py,sha256=od7uzON8z0OB9zPO7Jg_1pXZCufcQnWORKjABqSAgTw,545
55
+ dkist_processing_common/parsers/time.py,sha256=jcyhID_6ldkLZMBO9gqzYJ-oAD1kmFOThL7ASd7gCKs,8370
56
+ dkist_processing_common/parsers/unique_bud.py,sha256=ht_1Oi1OL7otZOMXcoVilmDSYVCC7KZ7Fj-TfKDFcqw,3173
57
+ dkist_processing_common/parsers/wavelength.py,sha256=Cb0opJmFRLQVPIMC5YRSDeygWjvQGS9avYrFhLsBekY,546
57
58
  dkist_processing_common/tasks/__init__.py,sha256=uH8DTiQP-cx4vMK53S4LYGZGmbip5s0kWORvZBrSNj8,624
58
59
  dkist_processing_common/tasks/assemble_movie.py,sha256=9K4sgXyRKaX7UsFBIs138pG3AtClwLLopYw3ZQY3ok4,12771
59
60
  dkist_processing_common/tasks/base.py,sha256=k_IJR5sVV6ennX0sbeb0C6dciqshdY7CKjtWHy_adm8,13143
60
61
  dkist_processing_common/tasks/l1_output_data.py,sha256=IM-nvGaTM5r-z-9vHr2wovPVUpuNCah-cWIFMO2fcII,10576
61
62
  dkist_processing_common/tasks/output_data_base.py,sha256=CC1TnCrChi8_iuMymr425CJqpY4jCggnVUMfqzFkpnw,3682
62
- dkist_processing_common/tasks/parse_l0_input_data.py,sha256=iRMGdvhxBobNsTDQ0IEl0myDfB4P_xpxA00guuBWDj8,7986
63
+ dkist_processing_common/tasks/parse_l0_input_data.py,sha256=N2FeeOagrRrl5Pi2y-U9iZS73t8rVPK0NtvZp5ukN7Y,7998
63
64
  dkist_processing_common/tasks/quality_metrics.py,sha256=dWuPKBD5elRCZEs4ZC91tyXxkuPUwMuvy9OoGU9Zs04,12489
64
65
  dkist_processing_common/tasks/teardown.py,sha256=e4LKnphJDYDVDAez2tH7MxpZgCmxYsKrq9Zk0qAkzzM,2355
65
66
  dkist_processing_common/tasks/transfer_input_data.py,sha256=8dDOfnT46qavGW-6fy-FT9LVb0TXANSpk1WpACpWK70,5787
66
67
  dkist_processing_common/tasks/trial_catalog.py,sha256=Y3DKstRfMS8nWWtJFMB0MUVPlZ1jWS_2jhJGMWwxy50,8748
67
68
  dkist_processing_common/tasks/trial_output_data.py,sha256=aI_aRuu0qVO8zFGrr_9baxx9i3jUEHZSmsmbO6ytlkE,6960
68
- dkist_processing_common/tasks/write_l1.py,sha256=EBQEViaLG2MzWsXOvNi8tsVsgzVJXBrLSGmYtsVgSlo,23781
69
+ dkist_processing_common/tasks/write_l1.py,sha256=EtCy7vltkZsZgoCAQybKgI35nmQncm0XrhIKxGJGSyc,22615
69
70
  dkist_processing_common/tasks/mixin/__init__.py,sha256=-g-DQbU7m1bclJYuFe3Yh757V-35GIDTbstardKQ7nU,68
70
71
  dkist_processing_common/tasks/mixin/globus.py,sha256=QAV8VElxMAqxJ2KSB_bJaraceovYfjHXjOdocrTCkIA,6592
71
72
  dkist_processing_common/tasks/mixin/interservice_bus.py,sha256=I7BUh0o8AEX-FZv7gxCts6is0uq9lycWjtTB2KqwBrU,1080
72
73
  dkist_processing_common/tasks/mixin/metadata_store.py,sha256=yTKijpQ-tNx_H2V_9HsZjMzkrzBDSQaSKkySV6VnnOM,11618
73
74
  dkist_processing_common/tasks/mixin/object_store.py,sha256=Vn4l2XuCimii9Fc3gM-pQGIkTKMv_ldqljlxkLesZLU,3236
74
75
  dkist_processing_common/tasks/mixin/quality/__init__.py,sha256=Bgu-DHW7yXLiehglldOCWluEkAP5qh0Hp1F30rh5NFw,383
75
- dkist_processing_common/tasks/mixin/quality/_base.py,sha256=nZ9IC-O-hsLXa5-tk29B13CZyQIdhJCv0eO9cdkAhWc,8303
76
- dkist_processing_common/tasks/mixin/quality/_metrics.py,sha256=lkHU6SeWeMfzf0QPM28_K9znllKx6rK91v1TiQ7vYfw,54556
76
+ dkist_processing_common/tasks/mixin/quality/_base.py,sha256=U1AEhj6OtF4YEdTkKWcgmoH6zrz6tYNjNXnmVU24L70,8491
77
+ dkist_processing_common/tasks/mixin/quality/_metrics.py,sha256=QUO6BWcFHZVc_QiXulGKnJk6cVzOalpFJq1Gusb7nZU,59844
77
78
  dkist_processing_common/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
78
79
  dkist_processing_common/tests/conftest.py,sha256=Rz1r2_by8aRZslSkS4AduEtpu3cpPxsAonZQyUCBPSQ,30867
79
80
  dkist_processing_common/tests/test_assemble_movie.py,sha256=XY_ruXSYP5k6s2gUAwlFdnhJ81eyWLSd2O9IkX4RXeo,4165
80
- dkist_processing_common/tests/test_assemble_quality.py,sha256=fWSHK4UdVqgNjvxQuD40NBUnXrtmthUP7PUbISPV4MQ,16897
81
+ dkist_processing_common/tests/test_assemble_quality.py,sha256=2fF2__iYj1cBShxxHMUP1sHHbDqDO8oqYICt6rg7lzo,17967
81
82
  dkist_processing_common/tests/test_base.py,sha256=4ST3__jEHitEQaQs9-0OcqtyEJfIjZsk_6PRYZFV2-U,7124
82
83
  dkist_processing_common/tests/test_codecs.py,sha256=FGhldrTdc28YD9FKrsW3lZ34LtvzecGP1qNi9fGHVGQ,22173
83
84
  dkist_processing_common/tests/test_constants.py,sha256=Kc9k5TdYy5QkRRlGav6kfI2dy5HHKqtpf9qOuaAfDZU,5903
84
85
  dkist_processing_common/tests/test_cs_step.py,sha256=RA0QD3D8eaL3YSOL_gIJ9wkngy14RQ2jbD-05KAziW4,2408
86
+ dkist_processing_common/tests/test_dkist_location.py,sha256=-_OoSw4SZDLFyIuOltHvM6PQjxm5hTiJQsiTGZ8Sadc,456
85
87
  dkist_processing_common/tests/test_fits_access.py,sha256=tyUPNbUqoTPhQgzua_doWP9l9ee4ir_LLV-I9rHktcs,11393
86
88
  dkist_processing_common/tests/test_flower_pot.py,sha256=X9_UI3maa3ZQncV3jYHgovWnawDsdEkEB5vw6EAB96o,3151
87
89
  dkist_processing_common/tests/test_fried_parameter.py,sha256=iXtlQIifZ6cDOkEi-YDgP3oAlss2loq08Uohgvy1byQ,1295
@@ -94,9 +96,9 @@ dkist_processing_common/tests/test_parameters.py,sha256=kNzX89vfrNRJ8d9rusMVv4DM
94
96
  dkist_processing_common/tests/test_parse_l0_input_data.py,sha256=SMNV1qyQTvnMx94MCNsiA-RyS9uxaxIABEDDxsuVzqY,10629
95
97
  dkist_processing_common/tests/test_publish_catalog_messages.py,sha256=wB2lcAnT77yVnqO0cFWOPxGf-tZ8U62kvvpiB5roBwQ,3268
96
98
  dkist_processing_common/tests/test_quality.py,sha256=vomy2YSPadKqJj2tG8sCs-UkQVvfKus7Cum7_Hpee4I,10257
97
- dkist_processing_common/tests/test_quality_mixin.py,sha256=hliXew6Sq7R166B2WZf022LyMKDio5G8jvkebHTm4ok,47067
99
+ dkist_processing_common/tests/test_quality_mixin.py,sha256=AFYLr6ZRWuxsAdPqvwwcgMfPhYCHKJRX-0oLJgoIDmg,55120
98
100
  dkist_processing_common/tests/test_scratch.py,sha256=7f28FMiSskSNX-bkRSrpJf2u1HQIbSvYajbjkeGMF9s,16481
99
- dkist_processing_common/tests/test_stems.py,sha256=DRTfSVC1bVJ8mD9E9eR1W5lUHXBR2uBNkYBcWnVlA5E,28471
101
+ dkist_processing_common/tests/test_stems.py,sha256=ini5dylLT5ioWKKWd1uu6kwx8Tf3aEnKKaGjf46a_GI,32649
100
102
  dkist_processing_common/tests/test_submit_dataset_metadata.py,sha256=F1IKBFWhjjMhONxEgs5p-cpjbQwxh7BLhDFw9b6j874,3806
101
103
  dkist_processing_common/tests/test_tags.py,sha256=UwlOJ45rkvbfbd5L5m5YltvOxQc8kGqJEn5V81H33U8,5023
102
104
  dkist_processing_common/tests/test_task_name.py,sha256=kqFr59XX2K87xzfTlClzDV4-Je1dx72LvdaJ22UE8UU,1233
@@ -107,7 +109,7 @@ dkist_processing_common/tests/test_transfer_l1_output_data.py,sha256=27PifkyH3RZ
107
109
  dkist_processing_common/tests/test_trial_catalog.py,sha256=SZ-nyn0MXU9Lkg_94FbKER_cwiGoi06GYlzF_3AmvKg,6802
108
110
  dkist_processing_common/tests/test_trial_output_data.py,sha256=cBCj0kXyF5NEMzKh6zPVksdoXyE8ju1opJgWgjdcJWA,12790
109
111
  dkist_processing_common/tests/test_workflow_task_base.py,sha256=Z5aPW5LQtS0UWJiYho4X0r-2gPLfzpkmMwfmaoFLjMg,10517
110
- dkist_processing_common/tests/test_write_l1.py,sha256=TKRSx1qX9shFlafP5H0tHy5ouIVzFLU0AMJOjViNygQ,27077
112
+ dkist_processing_common/tests/test_write_l1.py,sha256=U0Ge-sSvtKT4t5YrVochbwxSb1R_Lk-fKNbheDmhjtc,26709
111
113
  docs/Makefile,sha256=qnlVz6PuBqE39NfHWuUnHhNEA-EFgT2-WJNNNy9ttfk,4598
112
114
  docs/changelog.rst,sha256=S2jPASsWlQxSlAPqdvNrYvhk9k3FcFWNXFNDYXBSjl4,120
113
115
  docs/conf.py,sha256=FkX575cqTqZGCcLAjg2MlvE8Buj1Vt3CpHNgZxG256E,1890
@@ -116,7 +118,7 @@ docs/landing_page.rst,sha256=aPAuXFhBx73lEZ59B6E6JXxkK0LlxzD0n-HXqHrfumQ,746
116
118
  docs/make.bat,sha256=mBAhtURwhQ7yc95pqwJzlhqBSvRknr1aqZ5s8NKvdKs,4513
117
119
  docs/requirements.txt,sha256=Kbl_X4c7RQZw035YTeNB63We6I7pvXFU4T0Uflp2yDY,29
118
120
  licenses/LICENSE.rst,sha256=piZaQplkzOMmH1NXg6QIdo9wwo9pPCoHkvm2-DmH76E,1462
119
- dkist_processing_common-10.8.4rc1.dist-info/METADATA,sha256=mvmjsxz0nAMy1AaYE1etUD0u3QXFz4R2D3zaZca8csY,7150
120
- dkist_processing_common-10.8.4rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
121
- dkist_processing_common-10.8.4rc1.dist-info/top_level.txt,sha256=LJhd1W-Vn90K8HnQDIE4r52YDpUjjMWDnllAWHBByW0,48
122
- dkist_processing_common-10.8.4rc1.dist-info/RECORD,,
121
+ dkist_processing_common-11.0.0rc1.dist-info/METADATA,sha256=hXefUjocpqT4m0fISUEKWLeBLTPJrpSeCkK6SR4p37o,7207
122
+ dkist_processing_common-11.0.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
123
+ dkist_processing_common-11.0.0rc1.dist-info/top_level.txt,sha256=LJhd1W-Vn90K8HnQDIE4r52YDpUjjMWDnllAWHBByW0,48
124
+ dkist_processing_common-11.0.0rc1.dist-info/RECORD,,
changelog/235.feature.rst DELETED
@@ -1,3 +0,0 @@
1
- Add two new codecs: Basemodel codecs are used for encoding and decoding Pydantic BaseModel objects. For decoding, the intended model
2
- is passed to the decoder through a keyword argument in the task read method. Array codecs are used for encoding and decoding numpy
3
- arrays similar to the standard np.load() and np.save(), but with the task tag-based write method.
changelog/235.misc.1.rst DELETED
@@ -1,2 +0,0 @@
1
- Change the behavior of ParameterBase such that it takes the task scratch as an argument to provide access to the
2
- parameter document and parameters that are files. This behavior replaces the input dataset mixin parameter read method.
changelog/235.misc.rst DELETED
@@ -1 +0,0 @@
1
- Remove the input_dataset mixin and replace it with input_dataset Pydantic BaseModel models.