careamics 0.0.5__py3-none-any.whl → 0.0.6__py3-none-any.whl

{careamics-0.0.5.dist-info → careamics-0.0.6.dist-info}/RECORD

@@ -1,38 +1,43 @@
-careamics/__init__.py,sha256=xBCerWN66hv3T7dRGiUYLflmbJtJt1HqbSg9JCWp8pY,391
-careamics/careamist.py,sha256=jwR-idgdejBiQK7dJ968BjPsH18y9HTfu4UcYES4SIw,37588
+careamics/__init__.py,sha256=WF2JpQmC-MmuSB0L81XRo67NwaN_0qjyywcpRlbVJVE,569
+careamics/careamist.py,sha256=rakSaDSGRR0Pr0o1s8ejwIrjWknOQs4DgljukqfyWu0,37635
 careamics/conftest.py,sha256=Od4WcaaP0UP-XUMrFr_oo4e6c2hi_RvNbuaRTopwlmI,911
 careamics/py.typed,sha256=esB4cHc6c07uVkGtqf8at7ttEnprwRxwk8obY8Qumq4,187
 careamics/cli/__init__.py,sha256=LbM9bVtU1dy-khmdiIDXwvKy2v8wPBCEUuWqV_8rosA,106
-careamics/cli/conf.py,sha256=QuzGVTMHnqZqlnIqqNsgpcSgJUk0yNLr48hkKWiZF0E,13103
-careamics/cli/main.py,sha256=9b57xqiXchrHOw5_Ma0UZyD1hq3x2jslzL4HBIMvyvA,6593
-careamics/cli/utils.py,sha256=ndi6nVWenGCuikGSOcVTCoZQfU3RcAyuH5uUemQRK8M,667
-careamics/config/__init__.py,sha256=ywiA7k3OyDFlsRIPQ_4uc3yqzgEnxOtodD8V7OztB9Q,1188
+careamics/cli/conf.py,sha256=oixGRZNylW-NTM_rkDtQkSRw8KUYwtmUC_hK5BEeLnA,13074
+careamics/cli/main.py,sha256=S4B3c1ZN-OQK0l2_W42CaW0KmF_Pe_y4pKgn_UOuyDg,6564
+careamics/cli/utils.py,sha256=q_dmG7lxg_FT62qX9fPilIWL1M8ibhLnnhUKqa4knPI,660
+careamics/config/__init__.py,sha256=c5FXtcQrtROHdLRl4cHpKo6V4_E4yr6HxYgNwqH9CHg,1917
 careamics/config/callback_model.py,sha256=EeYHqpMIPQwyNxLRzzX32Uncl5mZuB1bJO76RHpNymg,4555
-careamics/config/configuration_factory.py,sha256=RHe3RyAg0YMN3B_gvAzODzzXQvuMY3X-Yyi6FgjCqW4,30400
-careamics/config/configuration_model.py,sha256=xJkrtBoLwRZOdL0mmEq08lciP1JH7i40UQHSPpEs1iI,18878
-careamics/config/data_model.py,sha256=MOXnX6dmI-08b4H_43xOrnSlwdbCtO1UuI9YqOT8djA,15370
-careamics/config/fcn_algorithm_model.py,sha256=eBBjkoyavhvDJFTrjJDkNrWSP_-YaK0cy7qo0lCdBQ0,5105
+careamics/config/care_configuration.py,sha256=hSfNJ-dooHm4ujG6Q3Hawr8zDeYy1HNiebtM7gxsh7s,2714
+careamics/config/configuration.py,sha256=KmLeXHkFhQTrcru1erhfVf3tHvQdoi12ls_u254rtDw,11114
+careamics/config/configuration_factories.py,sha256=-kdoLQ0dkV4xlcUx_odrtMsvL4NqGYg5-4HVl3PqY4c,32423
+careamics/config/configuration_io.py,sha256=ks9R8lRCBY_m0sdy1k4ZWFPKEFSp7K9X47jCG4d0FY4,2353
 careamics/config/inference_model.py,sha256=UE_-ZmCX6LFCbDBOwyGnvuAboF_JNX2m2LcF0WiwgCI,6961
-careamics/config/likelihood_model.py,sha256=4R80kLqIzMf_7JN6RaojLiOudykqp-KM1CuJpET7dho,2288
+careamics/config/likelihood_model.py,sha256=VorUtc0_-xIWNxwVrd1kBba-003ICdVMtxpcDCxH4Io,2259
 careamics/config/loss_model.py,sha256=yYcUBS90Qyon1MxeaHiVP3dJHPJFC0GUvWKGcAb3IHk,2036
-careamics/config/nm_model.py,sha256=xkQXrcXfMPAMpWoFCPRLyDfy4lP_-En4cAcOV4q21KI,4787
+careamics/config/n2n_configuration.py,sha256=QoHnEak0LiG6rBIXnmx61Chkz-Q0jr2Qt9I0NnmgZo4,2667
+careamics/config/n2v_configuration.py,sha256=5MagUFaVZ-9VGIiaKD-4RRkFpJen2imMoR3Lu07BD14,8489
+careamics/config/nm_model.py,sha256=5dAhDBLa4WPfKaNEK6ATNsSUwtlH8u8gYweEA4gZP6g,4758
 careamics/config/optimizer_models.py,sha256=OWpTydRBBR8wt_af1mZHNNwvL_RtnRFopAOdgjzLo30,5750
 careamics/config/tile_information.py,sha256=c-_xrVPOgcnjiEzQ-9A_GhNPamObkMANbeHaRP29R-4,2059
 careamics/config/training_model.py,sha256=67_ipo_-LxhT4-WqAs40Sg8PjU--my43Qn3BhjvlXxM,3212
-careamics/config/vae_algorithm_model.py,sha256=4oPeEI2vWhkQLOseX4iSAY_uy4NkVaOIxiXl5Tbh_eo,4653
-careamics/config/architectures/__init__.py,sha256=VyuBIopbqpR0KPq6cAeX3qPCrlbY8ZB_rGXwa2ApNvc,447
-careamics/config/architectures/architecture_model.py,sha256=qqPpmkNLwTBC3nwOmXpl33UAvpDFsAxFRBw9MIsQgws,921
-careamics/config/architectures/custom_model.py,sha256=-ROcKn_Ai7SZqzQLoEUc2BLAalgjyvoOnyDKUiY_cp0,4876
-careamics/config/architectures/lvae_model.py,sha256=62_GTtrh5_K0ijpRFh8mz26aaPkXShqtDnhh_Z0TfCQ,7566
-careamics/config/architectures/register_model.py,sha256=lHH0aUPmXtI3Bq_76zkhg07_Yb_nOJZkZJLCC_G-rZM,2434
-careamics/config/architectures/unet_model.py,sha256=8F2KosNkrXUP2bxlm-D1mowS9x3GOjyXjsEo1Kf-05k,3497
-careamics/config/references/__init__.py,sha256=rZAQzmrciX5cNICcXaBH6sbE6N6L7_qYQUkasNy9y-c,763
-careamics/config/references/algorithm_descriptions.py,sha256=wR3hIoeg5eiUEPbwTxMpQYLTKQyRl_5naSDbBZOZESU,3541
-careamics/config/references/references.py,sha256=AXx08FJQxHb7SYOluCr_eQn_mbOris5dXqhKrCnhBTE,1573
-careamics/config/support/__init__.py,sha256=pKqk76kyBraiSC1SQos-cyiQwsfOLLkLuWj6Hw60LZ4,1041
+careamics/config/algorithms/__init__.py,sha256=on5D6zBO9Lu-Tf5To8xpF6owIFqdN7RSmZdpyXDOaNw,404
+careamics/config/algorithms/care_algorithm_model.py,sha256=mdMDc0Vr0V7LZmksdjNDlRx3ofwy57IMXygY_vg6hPY,1175
+careamics/config/algorithms/n2n_algorithm_model.py,sha256=FO04TRecvMPikz_s2WupwODzi4g3x84g1HXRPgMAZWo,1015
+careamics/config/algorithms/n2v_algorithm_model.py,sha256=TBAUKeFxdRtomYhFrVqfq31hpOiNQO2YOogYzVVFknM,919
+careamics/config/algorithms/unet_algorithm_model.py,sha256=OaBFVlhsb9YhF3f2x1ImazvfnZ4_DPWvYWihRwurkeg,2587
+careamics/config/algorithms/vae_algorithm_model.py,sha256=1ZrShUGHA7zbjSiCQwhUSv7l7Hr7MbpcLOw8ucM27p8,4680
+careamics/config/architectures/__init__.py,sha256=lYUz56R7LDqVQWQDLkLgJL8VtOyxc_ege1r4bXGEBqA,220
+careamics/config/architectures/architecture_model.py,sha256=gXn4gdLrQP3bmTQxIhzkEHYlodPaIp6LI-kwZl23W-Y,911
+careamics/config/architectures/lvae_model.py,sha256=lwOiJYNUqPrZFl9SpPLYon77EiRbe2eI7pmpx45rO78,7606
+careamics/config/architectures/unet_model.py,sha256=HJJWf-wuYTv8KXMwikdjeB8htg1QOt0IyQbMuBt1LUI,3556
+careamics/config/data/__init__.py,sha256=ijFNSRKkKVF8fw6ym8kXq_wNhdwkWXxuNj2XKtD3KjE,218
+careamics/config/data/data_model.py,sha256=lSAIjpxuLaw1GMJ0rOHOZghYBwqWwlZf5sCphwAKMUc,11111
+careamics/config/data/n2v_data_model.py,sha256=-n5cncmcrd4-KUSmEk8rXAONMC7sfbYfGmshVNuySCU,5915
+careamics/config/support/__init__.py,sha256=ktWvxbTkRXnQPS_N84l9E2B5kTZVdd64SIjsJIQKB-k,1041
 careamics/config/support/supported_activations.py,sha256=CqOWoziIK5jZZXJO7G7cGg3TTid1POqv8FXqxjXxyME,535
-careamics/config/support/supported_algorithms.py,sha256=Tt5PaV1mwSZdrc4XpW9M4-ofExoHnMgfebCSN6ns0AQ,953
-careamics/config/support/supported_architectures.py,sha256=f93cRyeBOu5rYzvYrvWPVpjyK54GhyVTzCT1GNSdhI8,494
+careamics/config/support/supported_algorithms.py,sha256=ivLECrplTLjSUsa7AyT1n2aOo_WM3sMJJlW3OOT1fPk,846
+careamics/config/support/supported_architectures.py,sha256=pOxvHOAIUkc7HCO0IIg4K22h-Ti5ErtcIkGOjN-zh1s,340
 careamics/config/support/supported_data.py,sha256=T_mDiWLFMVji_EpjBABUObAJcnv-XBnqp9XUZP37Tdk,2902
 careamics/config/support/supported_loggers.py,sha256=ubSOkGoYabGbm_jmyc1R3eFcvcP-sHmuyiBi_d3_wLg,197
 careamics/config/support/supported_losses.py,sha256=2x5sZuxRbWJzodoL35I1mMYUUDMzk8UFiFdbyPwbJ4E,583
@@ -40,58 +45,58 @@ careamics/config/support/supported_optimizers.py,sha256=_2XmwzYENB6xpTedyWHUdWuG
 careamics/config/support/supported_pixel_manipulations.py,sha256=rFiktUlvoFU7s1NAKEMqsXOzLw5eaw9GtCKUznvq6xc,432
 careamics/config/support/supported_struct_axis.py,sha256=alZMA5Y-BpDymLPUEd1zqVY0xMkgl9Rv1d4ujED6sco,424
 careamics/config/support/supported_transforms.py,sha256=ODvmoTywvJWG_5-SJJZu-X1FNtKGhkNWQc-t26IFZWI,311
-careamics/config/transformations/__init__.py,sha256=lW9DcsCfemnlcmECyw5Id28nIuwg1YPKWhGgdfzcv5s,493
+careamics/config/transformations/__init__.py,sha256=jMTUX15n8ZF4Nc9gQp-qbXfhj-iEsEa65lzNbHwzyfY,631
 careamics/config/transformations/n2v_manipulate_model.py,sha256=Mdxc4J3vxe_dM2CIhmTwwGOIirQvrQXLoa2vRsTzoYI,1855
 careamics/config/transformations/normalize_model.py,sha256=1Rkk6IkF-7ytGU6HSzP-TpOi4RRWiQJ6fOd8zammXcg,1936
-careamics/config/transformations/transform_model.py,sha256=i7KAtSv4nah2H7uyJFKqg7RdKF68OHIPMNNvDo0HxGY,1000
-careamics/config/transformations/transform_union.py,sha256=3PnzsWlD2ymIgRPrUDZzPT9vMOVLVDtUcEorZTYx_-I,574
+careamics/config/transformations/transform_model.py,sha256=6UVbXnxm-LLZOQQ-ZBwWwgmS_99DiBuERLfMxrta3-8,990
+careamics/config/transformations/transform_unions.py,sha256=uqlI8Nm827bKfMbDQLVhKFtT9e7TJ_zIYDBdHlOuQ1I,1137
 careamics/config/transformations/xy_flip_model.py,sha256=zU-uZ1b1zNZWckbho3onN-B7BHKhN7jbgbNZyRQhv2s,1025
 careamics/config/transformations/xy_random_rotate90_model.py,sha256=6sYKmtCLvz0SV1qZgBSHUTH-CUjwvHnohq1HyPntbyE,894
 careamics/config/validators/__init__.py,sha256=iv0nVI0W7j9DxFPwh0DjRCzM9P8oLQn4Gwi5rfuFrrI,180
-careamics/config/validators/validator_utils.py,sha256=aNFzpBVbef3BZIt6MiNMVc2kW6MJDWqQgdYkFM8Gjig,2621
-careamics/dataset/__init__.py,sha256=NQSWdpQu6BhqGGHUYuOt1hXJrGUN1LPNCP1A8duMY84,547
-careamics/dataset/in_memory_dataset.py,sha256=7YRpbKg6nqrECDhaA88HNlstyTObQxTN9jPcNlE_aWE,9906
+careamics/config/validators/validator_utils.py,sha256=NVkEOr5AQK4JXWNtmgeQgAaJOyieJNb5PHCjlcqNeew,2611
+careamics/dataset/__init__.py,sha256=31vop67zbtGesENEIig-LLw1q2lCydMFc_YWgfK2Yt4,547
+careamics/dataset/in_memory_dataset.py,sha256=MV_Vf4siIP-g7VKhxN4rU7MZXpaHKvfwr8ZXqk44Qhs,9958
 careamics/dataset/in_memory_pred_dataset.py,sha256=VvwW5D8TjgO_kR8eZinP-9qepSiI6ZsUN7FZ0Rvc8Bs,2161
 careamics/dataset/in_memory_tiled_pred_dataset.py,sha256=DANmlnlV1ysXKdwGvmJoOYKcjlgoMhnSGSDRpeK79ZA,3552
-careamics/dataset/iterable_dataset.py,sha256=vHwkzoQs-CvbGHcGtvYMF52dO6zLau89A13xDOWSGUU,9770
-careamics/dataset/iterable_pred_dataset.py,sha256=2KC9C2hpZmhWSmo6w9Fhz0wjmbcsBlRy8QsYfO4dN2w,3740
-careamics/dataset/iterable_tiled_pred_dataset.py,sha256=uNpc_13vo9REvGYOLu7lBNDh813b_UqZ9x5c4Q_udDE,4533
+careamics/dataset/iterable_dataset.py,sha256=pqtm-AWhDbuZTnXf0roAHWVxGPRTekAzVcJHLzaSyFU,9797
+careamics/dataset/iterable_pred_dataset.py,sha256=jee4b8bZOyvSS5qfIsb6Jkk1EV_MKEU2SyZ0m7p0p9k,3767
+careamics/dataset/iterable_tiled_pred_dataset.py,sha256=2j_kLMB6DfSKXPszZPYgsB08TVgcf1V5HY_kZVozrFM,4560
 careamics/dataset/zarr_dataset.py,sha256=lojnK5bhiF1vyjuPtWXBrZ9sy5fT_rBvZJbbbnE-H_I,5665
-careamics/dataset/dataset_utils/__init__.py,sha256=b9r_2BcrXoHNq9chXfZvgINGwZRpWfUZ_p6vikB_Kxw,507
-careamics/dataset/dataset_utils/dataset_utils.py,sha256=zYNglet5lYKxIhTeOGG2K24oujC-m5zyYlwJcQcleVA,2662
-careamics/dataset/dataset_utils/file_utils.py,sha256=4Aq92wz9M7esrujDbOxw1WNoYLlEjBRa4sOzf2Aw61c,4070
-careamics/dataset/dataset_utils/iterate_over_files.py,sha256=TcX24NRt2cdM9gmmQV2f5ziwXxRne2-zePzz3DDFSMA,2871
+careamics/dataset/dataset_utils/__init__.py,sha256=MJ3xriL6R4ZtmzbvLsASUWLb85Hk5AdeRaYnHpNELJQ,507
+careamics/dataset/dataset_utils/dataset_utils.py,sha256=X83DzaOWmHdl4eOPac2IQJH3bPA43RVq0hPrFrzvIXQ,2630
+careamics/dataset/dataset_utils/file_utils.py,sha256=ru6AtQ9LCmo6raN1-GnJEN4UyP1PbmSdR9MEys3CuHo,4094
+careamics/dataset/dataset_utils/iterate_over_files.py,sha256=Jun35Qn9XevHOb_DixYBMHDAOykLmiwciA5Q2MzSUK8,2912
 careamics/dataset/dataset_utils/running_stats.py,sha256=kWorioMH4S5uZj2cvUpjHB6cIUhMFa1XXwDQrrKIWdI,5752
 careamics/dataset/patching/__init__.py,sha256=7-s12oUAZNlMOwSkxSwbD7vojQINWYFzn_4qIJ87WBg,37
 careamics/dataset/patching/patching.py,sha256=deAxY34Iz-mguBlHQ-5EO4vRhPpR9I3LQ9onV1K_KqA,8858
-careamics/dataset/patching/random_patching.py,sha256=61sLxA4eJN5TIWBVIDZdJahS_CkclpM7Kc_VdPj91dU,6486
-careamics/dataset/patching/sequential_patching.py,sha256=_l3Q2uYIhjMJMaxDdSbHC9_2kRF9eLz-Xs3r9i7j3Nc,5903
-careamics/dataset/patching/validate_patch_dimension.py,sha256=sQQ0-4b4uu60MNKkoWv95KxQ80J7Ku0CEk0-kAXlKeI,2134
-careamics/dataset/tiling/__init__.py,sha256=XynyAz85hVfkLtrG0lrMr_aBQm_YEwfu5uFcXMGHlOA,190
-careamics/dataset/tiling/collate_tiles.py,sha256=OrPZ-n-V3uGOc_7CcPnyEJqdbEVDlTfJfWmZnyBZ-HA,978
-careamics/dataset/tiling/lvae_tiled_patching.py,sha256=AOS_1m1q74YhfLgWeUx1xkRNvgw1dq6Vi7yJhmMR-A0,12992
-careamics/dataset/tiling/tiled_patching.py,sha256=ouxUWvttzmTY310CuiR25IGNVWgksVQAXoN4IXSi_G0,5942
-careamics/file_io/__init__.py,sha256=vdIx5JV3JwoyOeWnY_0tY6aekwPFy_8hBqe0Yj-aOH8,334
-careamics/file_io/read/__init__.py,sha256=I2Ios3fOoe_7f1nYT88qt2hcl0107aJCvA8yPfdpVIA,259
-careamics/file_io/read/get_func.py,sha256=yGXD0rTFD7u70FR0axrQtWies0aYW3iQ6f0Wfcd8z-8,1394
+careamics/dataset/patching/random_patching.py,sha256=gm1jxye9yvHbdijLzCtDSzRU_9j110GRLMnJaUwLAHQ,6487
+careamics/dataset/patching/sequential_patching.py,sha256=4F5E1Ta0M5kFXGwI2-QXRxeOx0CyUwbFaB5awkMCN_Q,5890
+careamics/dataset/patching/validate_patch_dimension.py,sha256=mC2bZWBpU44NEvXxEfR7ULUKwWPuZPjmBWpHYJxNDWc,2121
+careamics/dataset/tiling/__init__.py,sha256=aW_AMB9rzm0VmooUpjcyqv6sQP69RlPQMEdP2sVjdz8,190
+careamics/dataset/tiling/collate_tiles.py,sha256=XK0BsDQE7XwIwmOoCHJIpVC3kqjSN6nDhrJ4POVeHS8,965
+careamics/dataset/tiling/lvae_tiled_patching.py,sha256=LYEEdjKuKaxIGFtOkhfpsE7hruBnIsD5HcW9aVH6WHI,13019
+careamics/dataset/tiling/tiled_patching.py,sha256=6vxsqlccUqIl4Ys92JWIPs0Kn95VzaHoAYMSGcp2dh8,5956
+careamics/file_io/__init__.py,sha256=vgMI77X820VOWywAEW5W20FXfmbqBzx4V63D3V3_HhI,334
+careamics/file_io/read/__init__.py,sha256=wf8O_o80ghrlWQ-RGEuSqcc2LU55P1B-oxTacDToygo,259
+careamics/file_io/read/get_func.py,sha256=O_pdymjh2mc-JZ1je3ZnPAcsHc7Je3a005AMgAa0xuw,1388
 careamics/file_io/read/tiff.py,sha256=UMofW33rvByK9B1zYGhSrWAiAA3uQUV3OVK7cq9d0gQ,1359
 careamics/file_io/read/zarr.py,sha256=2jzREAnJDQSv0qmsL-v00BxmiZ_sp0ijq667LZSQ_hY,1685
-careamics/file_io/write/__init__.py,sha256=syy-e55OKPqa2Fn7G6szJrAmmJ4JUJyr4Y4ZSnKK0zg,283
+careamics/file_io/write/__init__.py,sha256=CUt33cRjG9hm18L9a7XqaUKWQ_3xiuQ9ztz4Ab7RYG0,283
 careamics/file_io/write/get_func.py,sha256=hyGHe1RX-lfa9QFAnwRCz_gS0NRiRnXEtg4Bdeh2Esc,1627
 careamics/file_io/write/tiff.py,sha256=tBGIgl-I1sMyBivgx-dOTBykXBODkgwPH8MT3_4KAE8,1050
-careamics/lightning/__init__.py,sha256=iS9dYpYrkjyeZfodcohc25rBTPzxj9l50iTb5Jv0j0o,588
-careamics/lightning/lightning_module.py,sha256=1VlPDefLbDZggzZ4lZ68s2xi-thwS1MGTtD37RBifT0,22466
+careamics/lightning/__init__.py,sha256=ATCVAGnX08Ik4TxbIv0-cXb52UinR42JgvZh_GIMSpc,588
+careamics/lightning/lightning_module.py,sha256=3nzRcjv7dlWmWjoWmGaBV7D6d698xMp3XO7fpIrZhwA,22630
 careamics/lightning/predict_data_module.py,sha256=JNwujK6QwObSx6P25ghpGl2f2gGT3KVgYMTlonZzH20,12745
-careamics/lightning/train_data_module.py,sha256=prgU3d_MTBsFXZM_XNaLeXuFeSmgrhVxDk2wJwaCzaM,28512
-careamics/lightning/callbacks/__init__.py,sha256=neTWqF6EbBRVf0FqtBkPHeGIR3j1yTk0OASraEVN0Pg,312
+careamics/lightning/train_data_module.py,sha256=7fti07y2TTV_Airl-D9FRTJf7AmsFmErSKnWUGLQE58,28699
+careamics/lightning/callbacks/__init__.py,sha256=eA5ltzYNzuO0uMEr1jG4wP01b0s29s5I03WGJ290qkw,312
 careamics/lightning/callbacks/hyperparameters_callback.py,sha256=u45knOZHwoVHz6yYfrnERQuozT_SfZ1OrKP0QjeU4EM,1495
-careamics/lightning/callbacks/progress_bar_callback.py,sha256=8HvNSWZldixd6pjz0dLDo0apIbzTovv5smKmZ6tZQ8U,2444
-careamics/lightning/callbacks/prediction_writer_callback/__init__.py,sha256=tATV4kFZ8h4ZpN866URsf2vbEfY9HlHN4VALJcDySCY,548
+careamics/lightning/callbacks/progress_bar_callback.py,sha256=RilGAVUa90AlCLdooIGJF2cAcHAjGn24zKZiBmRRkwg,2438
+careamics/lightning/callbacks/prediction_writer_callback/__init__.py,sha256=ZVf3vaSU_NjSjrKbI24H0kK9WAiP9oKXfhP670EaWMo,548
 careamics/lightning/callbacks/prediction_writer_callback/file_path_utils.py,sha256=i4vGGiVLslafi-5iuvkAKzBgZ0BpwTTxSTo31oViFz4,1480
-careamics/lightning/callbacks/prediction_writer_callback/prediction_writer_callback.py,sha256=HGDyYLSak9puGEs42cI_08peAuOMaFyZaRkcpv_n0iY,8183
-careamics/lightning/callbacks/prediction_writer_callback/write_strategy.py,sha256=aqc6DDewD4n3tbLfqS4y-FfNKV9zy3-CZ0K2Mu16Mms,12567
+careamics/lightning/callbacks/prediction_writer_callback/prediction_writer_callback.py,sha256=8HHUSKcG7G0FSCVPnpGQHLfpara5mnKAwsiiyWp2wzo,8210
+careamics/lightning/callbacks/prediction_writer_callback/write_strategy.py,sha256=lxsLjLskRpYnzdyWCdOICUJxF9YzuUi1RH0LJnOCVgo,12594
 careamics/lightning/callbacks/prediction_writer_callback/write_strategy_factory.py,sha256=F1IpbNNgkv5eK8Xpqp7wqv2lsqEdP1wMRlBL7RBn93U,7114
-careamics/losses/__init__.py,sha256=4s73OlMCGJdNCVfOteJPyaPwflgIiu_fj3zpKLx7j6o,351
+careamics/losses/__init__.py,sha256=nSWbkBcFhkyUkIT2wVcULqpieyY2Oro39NXZTtfQpXo,351
 careamics/losses/loss_factory.py,sha256=oPacrkwiabsmiW_r--IxX-XPRbzezZUvOuWKbUw5LiI,1518
 careamics/losses/fcn/__init__.py,sha256=kf92MKFGHr6upiztZVgWwtGPf734DZyub92Rn8uEq8o,18
 careamics/losses/fcn/losses.py,sha256=NdOz29hzJ7D26p13q-g0NWoYwNauIWrP2xWww6YPbB8,2360
@@ -117,55 +122,55 @@ careamics/lvae_training/dataset/utils/data_utils.py,sha256=8PvRPqSbYHPCl87cycZHx
 careamics/lvae_training/dataset/utils/empty_patch_fetcher.py,sha256=OFjeqhZ6vFULJsF5tnoByhEhE8aLHujFToU_yyqMCP4,2266
 careamics/lvae_training/dataset/utils/index_manager.py,sha256=Gt1I-7lBaQDBgqguOmofAFDdAsJQfz7ktvq4_I80F9c,10084
 careamics/lvae_training/dataset/utils/index_switcher.py,sha256=ZoMi8LsaIkm8MFqIFaxN4oQGyzCcwOlCom8SYNus15E,6716
-careamics/model_io/__init__.py,sha256=HITzjiuZQwo-rQ2_Ma3bz9l7PDANv1_S489E-tffV9s,155
-careamics/model_io/bmz_io.py,sha256=P0fMebLPFPQCuznS8qHfIuLFrRbsRM_SaXhWFq0nVRQ,8138
-careamics/model_io/model_io_utils.py,sha256=Dc_0YmcUfS3HuEVgN7KFfbDH8SGywT_eUCIJDVpEUHc,2755
-careamics/model_io/bioimage/__init__.py,sha256=r94nu8WDAvj0Fbu4C-iJXdOhfSQXeZBvN3UKsLG0RNI,298
-careamics/model_io/bioimage/_readme_factory.py,sha256=hpHtnKtousPthBESs3gPcbEytT2M6KAeD5uxzcKr280,3434
+careamics/model_io/__init__.py,sha256=khMIkk107LL5JGze0OVfl5Lfi14R3_e4W21tW0iJ1kE,155
+careamics/model_io/bmz_io.py,sha256=0fL0cz_80nsu_Jgk-ImXKTNbn_WSdrXB9flvLm2g3Is,7792
+careamics/model_io/model_io_utils.py,sha256=HCSoNvaOo55kI7teZleN57riqNK2fLq77zMadBKXCyc,2777
+careamics/model_io/bioimage/__init__.py,sha256=dKm-UluVwa_7EHQB0ukx-Qk8JVWtuT-OUinY8hE9EIw,298
+careamics/model_io/bioimage/_readme_factory.py,sha256=sRfHbuwhfYmpwsG0keXFCSK3qCS4VI4GQhX6OhSKLjY,3440
 careamics/model_io/bioimage/bioimage_utils.py,sha256=YVr75SDfafOiuYGonPbcsO-xVS0wI1WkmWQQZx6DXYQ,1246
 careamics/model_io/bioimage/cover_factory.py,sha256=8URrpEfJvdHBJeSrh5H2IQHSUybsTyAOR3_A-YYAAlw,4583
-careamics/model_io/bioimage/model_description.py,sha256=0swY2p5OHsc9Npfv4KQE4hDdh-1_n5eITfG41FM1vF0,10063
+careamics/model_io/bioimage/model_description.py,sha256=HwjWT77nIt6LkLvIBRxwgPOsL0ffrCMtDUjim39qgqE,10071
 careamics/models/__init__.py,sha256=Xui2BLJd1I2r_E3Sj24fJALFTi2FGtfNscUWj_0c9Hk,93
 careamics/models/activation.py,sha256=nu3sDgd7Lsyw8rvmUwxNN-7SM09cEMfxZ9DRDzdSKns,1049
-careamics/models/layers.py,sha256=_rSCMu0x6O88LeJL-dKkqPFp5A8aGD6s9ZQ7KhDRy-o,13770
-careamics/models/model_factory.py,sha256=hqhV8sDq1JBLKt_7Vrw4wJSugKmBt2FIWyO7ePnJSTo,1759
-careamics/models/unet.py,sha256=3pXpiCIw7WUaDV0Jmczkxi99C5-Zu3NpQpWxgRkeGL8,14321
+careamics/models/layers.py,sha256=tpsxbolRWYycZGxS3hKlDRtMf6HpNdZs98uwx5K8lls,13757
+careamics/models/model_factory.py,sha256=GWbouERvEfHj_BrpKHYgrPAj8dpdoh1R-X--Jt09N1Q,1370
+careamics/models/unet.py,sha256=9m8GxsTXX9c0mC-eVe2ZXQn2afose1CG6Z8vIhELb7I,14308
 careamics/models/lvae/__init__.py,sha256=6dT6uqgT__V08EjoTGxXguTbTkySZmByS9J2Bj6WWLM,53
 careamics/models/lvae/layers.py,sha256=UPxZiZjgnBnPs_wdSzcP-_s17MEw4P1CoIZzn4OdUA0,57944
 careamics/models/lvae/likelihoods.py,sha256=SHsaZZTjGg8TO6JkvjdQOZtu8CCOvlSxtqRBZhrGXKk,12098
-careamics/models/lvae/lvae.py,sha256=B-GZegbvCJ_cGwQ49hLtE9OIKPUxSyd3SsBfb8llHfs,34303
+careamics/models/lvae/lvae.py,sha256=AWll857rAnlvjTom5CX0CpbEcf-ci_Icoz5wFKVLu5E,34220
 careamics/models/lvae/noise_models.py,sha256=dFWM9DDJ7qzsyiT8sDWR8THbEiZmu3XnW5UzGJl7Mck,21834
 careamics/models/lvae/stochastic.py,sha256=019M6BBR6GtYjUVF6pcOTOsfEAYeg0vclz55V6Fl4yY,16588
 careamics/models/lvae/utils.py,sha256=EE3paHu3vhCaqfOrGypzUsImZJO94uBhx8q6kZ-R36o,11516
-careamics/prediction_utils/__init__.py,sha256=uYKzirlF-unFL9GbDPxFnYgOwSjGAtik9fonU7DfuEY,270
+careamics/prediction_utils/__init__.py,sha256=k5hsPGY8FOkwIT0fQgrUz7fVCH2NlwuOdZiISdXjEWg,270
 careamics/prediction_utils/lvae_prediction.py,sha256=ZwPFCSeUGsULIMoMQWRYKHfLFaDm7UKyGaUMVfSUqfs,6210
 careamics/prediction_utils/lvae_tiling_manager.py,sha256=SI-JaJvLrKWBSHdm-FjcqWdbhlcflTRiKxYF7CSGzvA,13736
-careamics/prediction_utils/prediction_outputs.py,sha256=p3Nbw9wRLU_M5uixPbzj_DmfxcSL7Y8rAQ_aTx70KQI,4082
-careamics/prediction_utils/stitch_prediction.py,sha256=HlfkJDirzbmil6Db-kFzlmhUREmGEZ2Ag8g-URbdIb0,3899
-careamics/transforms/__init__.py,sha256=VIHIsC8sMAh1TCm67ifB816Zp-LRo6rAONPuT2Qs3bs,483
-careamics/transforms/compose.py,sha256=EOEsga7oGNZZW8zerOz7TgirbLpGyaLf1WEnTXlRjoo,5677
-careamics/transforms/n2v_manipulate.py,sha256=op-BT3LJaHgHULuqRRqTI6e45FtvvUZ0y-Smzbrh2WU,5693
+careamics/prediction_utils/prediction_outputs.py,sha256=fw-bJ2szWJD7BgZlECmxy5sgeXGFJl4T8cRNzLR1aUQ,4069
+careamics/prediction_utils/stitch_prediction.py,sha256=8YRW2rea-is5tYI0Q1bw3bpX7VMFmbpxSP_y6x9Yfug,3893
+careamics/transforms/__init__.py,sha256=WtgpSFL_CJwpa47XzqS7bVXHPJ4qW0TamEymy_kgWQQ,483
+careamics/transforms/compose.py,sha256=ZSVwKg3LT2PrwtSBKtkb6AHsVSSlZIdv9wTYl4To1s4,5682
+careamics/transforms/n2v_manipulate.py,sha256=t9rtMbYV6P1IVp4yzuJfq5-giWyfGrxL8ZhzP29Pp8k,5686
 careamics/transforms/normalize.py,sha256=fxs813ydCWrIzrxFzkbk1gW8OGSr0esQSrNUFSJuGL0,7715
-careamics/transforms/pixel_manipulation.py,sha256=sMR3A1GrgL5AactzZYzvvtz7L13sA34ckVEoNTBmDqM,13362
+careamics/transforms/pixel_manipulation.py,sha256=38NsxY8ARvz7GSNDKx5g67Hv5qciBzadiELAg5OcUSU,13355
 careamics/transforms/struct_mask_parameters.py,sha256=jE29Li9sx3olaRnqYfJsSlKi2t0WQzJmCm9aCbIQEsA,421
 careamics/transforms/transform.py,sha256=cEqc4ci8na70i-HIGYC7udRfVa8D_8OjdRVrr3txLvQ,464
 careamics/transforms/tta.py,sha256=78S7Df9rLHmEVSQSI1qDcRrRJGauyG3oaIrXkckCkmw,2335
 careamics/transforms/xy_flip.py,sha256=64BDo8bmAEwO1TNhbIYcUJPzzVmY5ZyNaSNmmGLkn0U,3842
-careamics/transforms/xy_random_rotate90.py,sha256=0SsCTPGlpt-VCJvmr55KRULp3dwFMBN8fTDboJcEbWg,3192
-careamics/utils/__init__.py,sha256=rG_dnqX7rdyNTFWlDkIdNtDwwMQBpg_ym14ZFeYrWfs,402
+careamics/transforms/xy_random_rotate90.py,sha256=Kin42yaV4Z8lOwC9nN8gxK73rgnJ2MhCoHHPQmlSgvc,3185
+careamics/utils/__init__.py,sha256=mLwBQ7wTL2EwDwL3NcX53EHPNklojU45Jcc728y4EWQ,402
 careamics/utils/autocorrelation.py,sha256=M_WYzrEOQngc5iSXWar4S3-EOnK6DfYHPC2vVMeu_Bs,945
 careamics/utils/base_enum.py,sha256=bz1D8mDx5V5hdnJ3WAzJXWHJTbgwAky5FprUt9F5cMA,1387
-careamics/utils/context.py,sha256=Ljf70OR1FcYpsVpxb5Sr2fzmPVIZgDS1uZob_3BcELg,1409
+careamics/utils/context.py,sha256=SoTZfzG6fO4SDOGHOTL2Xlm1n1CSgb9B57GVhrEkFls,1436
 careamics/utils/lightning_utils.py,sha256=DMMmqx-AlNtddBCqm8b_W3h09qUetz32OMPhdDieFwg,1769
-careamics/utils/logging.py,sha256=coIscjkDYpqcsGnsONuYOdIYd6_gHxdnYIZ-e9Y2Ybg,10322
+careamics/utils/logging.py,sha256=5U4VsQ4m4OajtirLH6qUjrM1CAc-oXeCsd6JyROjkWE,10337
 careamics/utils/metrics.py,sha256=yAoCvrZ1kQx-kT9xdTBYz-oh0I52ef6uBnw8qgzpwn8,10318
 careamics/utils/path_utils.py,sha256=8AugiG5DOmzgSnTCJI8vypXaPE0XhnR-9pzeiFUZ-0I,554
 careamics/utils/ram.py,sha256=tksyn8dVX_iJXmrDZDGub32hFZWIaNxnMheO5G1p43I,244
 careamics/utils/receptive_field.py,sha256=Y2h4c8S6glX3qcx5KHDmO17Kkuyey9voxfoXyqcAfiM,3296
 careamics/utils/serializers.py,sha256=mILUhz75IMpGKnEzcYu9hlOPG8YIiIW09fk6eZM7Y8k,1427
-careamics/utils/torch_utils.py,sha256=dmRHalwp8F5lnQIkJzpt3qhHXFi8anD4OTPr9-x9jf4,3016
-careamics-0.0.5.dist-info/METADATA,sha256=3Q2mgo1N_eoKlR_xF-6eHAu3Vqff8Ncp_FAtpRPUDC0,3874
-careamics-0.0.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-careamics-0.0.5.dist-info/entry_points.txt,sha256=2fSNVXJWDJgFLATVj7MkjFNvpl53amG8tUzC3jf7G1s,53
-careamics-0.0.5.dist-info/licenses/LICENSE,sha256=6zdNW-k_xHRKYWUf9tDI_ZplUciFHyj0g16DYuZ2udw,1509
-careamics-0.0.5.dist-info/RECORD,,
+careamics/utils/torch_utils.py,sha256=_Cf3HdlIRl5hxfpUg9aofCSlcW7GSsIJxsbSORXko0U,3010
+careamics-0.0.6.dist-info/METADATA,sha256=Hz6RQlh5szAGllr7baFnc86U_sSC2fCjvU4FhjmhYMk,3898
+careamics-0.0.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+careamics-0.0.6.dist-info/entry_points.txt,sha256=2fSNVXJWDJgFLATVj7MkjFNvpl53amG8tUzC3jf7G1s,53
+careamics-0.0.6.dist-info/licenses/LICENSE,sha256=6zdNW-k_xHRKYWUf9tDI_ZplUciFHyj0g16DYuZ2udw,1509
+careamics-0.0.6.dist-info/RECORD,,

careamics/config/architectures/custom_model.py

@@ -1,162 +0,0 @@
-"""Custom architecture Pydantic model."""
-
-from __future__ import annotations
-
-import inspect
-from pprint import pformat
-from typing import Any, Literal
-
-from pydantic import ConfigDict, field_validator, model_validator
-from torch.nn import Module
-from typing_extensions import Self
-
-from .architecture_model import ArchitectureModel
-from .register_model import get_custom_model
-
-
-class CustomModel(ArchitectureModel):
-    """Custom model configuration.
-
-    This Pydantic model allows storing parameters for a custom model. In order for the
-    model to be valid, the specific model needs to be registered using the
-    `register_model` decorator, and its name correctly passed to this model
-    configuration (see Examples).
-
-    Attributes
-    ----------
-    architecture : Literal["custom"]
-        Discriminator for the custom model, must be set to "custom".
-    name : str
-        Name of the custom model.
-    parameters : CustomParametersModel
-        All parameters, required for the initialization of the torch module have to be
-        passed here.
-
-    Raises
-    ------
-    ValueError
-        If the custom model `name` is unknown.
-    ValueError
-        If the custom model is not a torch Module subclass.
-    ValueError
-        If the custom model parameters are not valid.
-
-    Examples
-    --------
-    >>> from torch import nn, ones
-    >>> from careamics.config import CustomModel, register_model
-    >>> # Register a custom model
-    >>> @register_model(name="my_linear")
-    ... class LinearModel(nn.Module):
-    ...    def __init__(self, in_features, out_features, *args, **kwargs):
-    ...        super().__init__()
-    ...        self.in_features = in_features
-    ...        self.out_features = out_features
-    ...        self.weight = nn.Parameter(ones(in_features, out_features))
-    ...        self.bias = nn.Parameter(ones(out_features))
-    ...    def forward(self, input):
-    ...        return (input @ self.weight) + self.bias
-    ...
-    >>> # Create a configuration
-    >>> config_dict = {
-    ...     "architecture": "custom",
-    ...     "name": "my_linear",
-    ...     "in_features": 10,
-    ...     "out_features": 5,
-    ... }
-    >>> config = CustomModel(**config_dict)
-    """
-
-    # pydantic model config
-    model_config = ConfigDict(
-        extra="allow",
-    )
-
-    # discriminator used for choosing the pydantic model in Model
-    architecture: Literal["custom"]
-    """Name of the architecture."""
-
-    name: str
-    """Name of the custom model."""
-
-    @field_validator("name")
-    @classmethod
-    def custom_model_is_known(cls, value: str) -> str:
-        """Check whether the custom model is known.
-
-        Parameters
-        ----------
-        value : str
-            Name of the custom model as registered using the `@register_model`
-            decorator.
-
-        Returns
-        -------
-        str
-            The custom model name.
-        """
-        # delegate error to get_custom_model
-        model = get_custom_model(value)
-
-        # check if it is a torch Module subclass
-        if not issubclass(model, Module):
-            raise ValueError(
-                f'Retrieved class {model} with name "{value}" is not a '
-                f"torch.nn.Module subclass."
-            )
-
-        return value
-
-    @model_validator(mode="after")
-    def check_parameters(self: Self) -> Self:
-        """Validate model by instantiating the model with the parameters.
-
-        Returns
-        -------
-        Self
-            The validated model.
-        """
-        # instantiate model
-        try:
-            get_custom_model(self.name)(**self.model_dump())
-        except Exception as e:
-            raise ValueError(
-                f"while passing parameters to the model {e}. Verify that all "
-                f"mandatory parameters are provided, and that either the {e} accepts "
-                f"*args and **kwargs in its __init__() method, or that no additional"
-                f"parameter is provided. Trace: "
-                f"filename: {inspect.trace()[-1].filename}, function: "
-                f"{inspect.trace()[-1].function}, line: {inspect.trace()[-1].lineno}"
-            ) from None
-
-        return self
-
-    def __str__(self) -> str:
-        """Pretty string representing the configuration.
-
-        Returns
-        -------
-        str
-            Pretty string.
-        """
-        return pformat(self.model_dump())
-
-    def model_dump(self, **kwargs: Any) -> dict[str, Any]:
-        """Dump the model configuration.
-
-        Parameters
-        ----------
-        **kwargs : Any
-            Additional keyword arguments from Pydantic BaseModel model_dump method.
-
-        Returns
-        -------
-        dict[str, Any]
-            Model configuration.
-        """
-        model_dict = super().model_dump()
-
-        # remove the name key
-        model_dict.pop("name")
-
-        return model_dict

careamics/config/architectures/register_model.py

@@ -1,103 +0,0 @@
-"""Custom model registration utilities."""
-
-from typing import Callable
-
-from torch.nn import Module
-
-CUSTOM_MODELS = {}  # dictionary of custom models {"name": __class__}
-
-
-def register_model(name: str) -> Callable:
-    """Decorator used to register a torch.nn.Module class with a given `name`.
-
-    Parameters
-    ----------
-    name : str
-        Name of the model.
-
-    Returns
-    -------
-    Callable
-        Function allowing to instantiate the wrapped Module class.
-
-    Raises
-    ------
-    ValueError
-        If a model is already registered with that name.
-
-    Examples
-    --------
-    ```python
-    @register_model(name="linear")
-    class LinearModel(nn.Module):
-        def __init__(self, in_features, out_features):
-            super().__init__()
-
-            self.weight = nn.Parameter(ones(in_features, out_features))
-            self.bias = nn.Parameter(ones(out_features))
-
-        def forward(self, input):
-            return (input @ self.weight) + self.bias
-    ```
-    """
-    if name is None or name == "":
-        raise ValueError("Model name cannot be empty.")
-
-    if name in CUSTOM_MODELS:
-        raise ValueError(
-            f"Model {name} already exists. Choose a different name or run "
-            f"`clear_custom_models()` to empty the registry."
-        )
-
-    def add_custom_model(model: Module) -> Module:
-        """Add a custom model to the registry and return it.
-
-        Parameters
-        ----------
-        model : Module
-            Module class to register.
-
-        Returns
-        -------
-        Module
-            The registered model.
-        """
-        # add model to the registry
-        CUSTOM_MODELS[name] = model
-
-        return model
-
-    return add_custom_model
-
-
-def get_custom_model(name: str) -> Module:
-    """Get the custom model corresponding to `name` from the registry.
-
-    Parameters
-    ----------
-    name : str
-        Name of the model to retrieve.
-
-    Returns
-    -------
-    Module
-        The requested model.
-
-    Raises
-    ------
-    ValueError
-        If the model is not registered.
-    """
-    if name not in CUSTOM_MODELS:
-        raise ValueError(
-            f"Model {name} is unknown. Have you registered it using "
-            f'@register_model("{name}") as decorator?'
-        )
-
-    return CUSTOM_MODELS[name]
-
-
-def clear_custom_models() -> None:
-    """Clear the custom models registry."""
-    # clear dictionary
-    CUSTOM_MODELS.clear()