ob-metaflow-stubs 6.0.3.143rc0__py2.py3-none-any.whl → 6.0.3.144__py2.py3-none-any.whl

metaflow-stubs/runner/utils.pyi

@@ -1,15 +1,15 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.062755                                                            #
+# MF version: 2.14.2.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-21T20:43:13.427631                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.runner.click_api
     import metaflow.runner.subprocess_manager
+    import metaflow.runner.click_api
     import typing
 
 

metaflow-stubs/system/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.052210                                                            #
+# MF version: 2.14.2.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-21T20:43:13.416464                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/system/system_logger.pyi

@@ -1,15 +1,15 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.078546                                                            #
+# MF version: 2.14.2.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-21T20:43:13.443504                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import typing
     import metaflow.event_logger
+    import typing
 
 
 class SystemLogger(object, metaclass=type):

metaflow-stubs/system/system_monitor.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.078199                                                            #
+# MF version: 2.14.2.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-21T20:43:13.443074                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/tagging_util.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.034445                                                            #
+# MF version: 2.14.2.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-21T20:43:13.398740                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/tuple_util.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.038644                                                            #
+# MF version: 2.14.2.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-21T20:43:13.403043                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/user_configs/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.034138                                                            #
+# MF version: 2.14.2.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-21T20:43:13.398457                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/user_configs/config_decorators.pyi

@@ -1,20 +1,20 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.038474                                                            #
+# MF version: 2.14.2.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-21T20:43:13.402864                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.user_configs.config_parameters
-    import metaflow.parameters
-    import metaflow.user_configs.config_decorators
     import metaflow.flowspec
-    import functools
+    import metaflow.user_configs.config_parameters
     import typing
     import metaflow.decorators
+    import metaflow.parameters
+    import functools
+    import metaflow.user_configs.config_decorators
 
 from ..exception import MetaflowException as MetaflowException
 from .config_parameters import ConfigValue as ConfigValue

metaflow-stubs/user_configs/config_options.pyi

@@ -1,13 +1,13 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.055177                                                            #
+# MF version: 2.14.2.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-21T20:43:13.419960                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import typing
 import metaflow
+import typing
 if typing.TYPE_CHECKING:
     import metaflow._vendor.click.types
     import typing

metaflow-stubs/user_configs/config_parameters.pyi

@@ -1,21 +1,21 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.036996                                                            #
+# MF version: 2.14.2.1+obcheckpoint(0.1.8);ob(v1)                                                    #
+# Generated on 2025-02-21T20:43:13.401381                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import collections
+import metaflow
 import abc
 import typing
-import metaflow
+import collections
 if typing.TYPE_CHECKING:
     import metaflow.user_configs.config_parameters
+    import typing
     import metaflow.parameters
-    import collections.abc
     import abc
-    import typing
+    import collections.abc
 
 from ..exception import MetaflowException as MetaflowException
 from ..parameters import Parameter as Parameter

{ob_metaflow_stubs-6.0.3.143rc0.dist-info → ob_metaflow_stubs-6.0.3.144.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-stubs
-Version: 6.0.3.143rc0
+Version: 6.0.3.144
 Summary: Metaflow Stubs: Stubs for the metaflow package
 Author: Netflix, Outerbounds & the Metaflow Community
 Author-email: help@outerbounds.co

ob_metaflow_stubs-6.0.3.144.dist-info/RECORD

@@ -0,0 +1,197 @@
+metaflow-stubs/__init__.pyi,sha256=2jOXs24ihjcABaA8Ngk8bk8ajqZf3GDwzCFt1rIkX_o,64857
+metaflow-stubs/cards.pyi,sha256=Ur__QzYKD3Ka9niHcBLGHjDn25aLsA8NFPE4_GgTcOg,1640
+metaflow-stubs/cli.pyi,sha256=sUr99hyvTSuVs8HM9YymHdGp5TgLbd2JqABSYUPe6RQ,2170
+metaflow-stubs/events.pyi,sha256=BFiJ2l7GeLvEaRxP90BGrqDZ-gpkp3RVFThM-r89wu4,3974
+metaflow-stubs/exception.pyi,sha256=3QKGaUYF49VAECV_OX-g_OHO6j5P3lOMJl44oOgibvc,2661
+metaflow-stubs/flowspec.pyi,sha256=yBA-eHb0Bq90guFLwXgr0wehOBOAGPif9l7KWN_D8G4,10327
+metaflow-stubs/generated_for.txt,sha256=1vXZAx-O2YEH0LPomZBZNvEAuD_hzArlUc8K039M0AA,62
+metaflow-stubs/includefile.pyi,sha256=HBVhC_RVg5Bs6BIo9G6n1X4mw4ZE_2O6C456iE9Y3DQ,4615
+metaflow-stubs/info_file.pyi,sha256=YDREgnuz2GBQTQMvXv3AxWEoa8DmR6oUqbaaFCzlPew,625
+metaflow-stubs/metaflow_config.pyi,sha256=EdVZWyfGXMP6rxweNN-XpXrU7F1kDLj1xK2N8zzZA8c,4999
+metaflow-stubs/metaflow_current.pyi,sha256=ULeplOYWqWjdlA6HXYdmQVT2dkx5z674V9tBQSuqpXo,18083
+metaflow-stubs/multicore_utils.pyi,sha256=7WOj_NEATcvnzZ0JmgJshUC3IOeW9MJpuNjy_joaAzM,2516
+metaflow-stubs/parameters.pyi,sha256=LJdYOZUZd1LpYRsFSlJmItA7LhPEpodMWFr6_slVC3s,7413
+metaflow-stubs/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+metaflow-stubs/pylint_wrapper.pyi,sha256=2nfrwgun1vV1EGOGJDR2ZPgO5DOelRnTAdaFE2D_zVI,994
+metaflow-stubs/tagging_util.pyi,sha256=U9sk6B9slPZAgAkb0B84GfVm2zcoX6z4nCoTyuQquos,1494
+metaflow-stubs/tuple_util.pyi,sha256=LZQhpMV-xbEq9pRKq8QJFAp8D03JQnG3JHZtC7irltE,663
+metaflow-stubs/cli_components/__init__.pyi,sha256=s8updti0zrOYUR1IPg3bBdf3fOGt1Gz1kkUlK4lnsTs,583
+metaflow-stubs/cli_components/utils.pyi,sha256=jdFTy4Z7G4rJs2dP9mLB0FruK_FyDW5siwoivvaP7Po,1260
+metaflow-stubs/client/__init__.pyi,sha256=HdDCSnoVZ4JJ6HdFbJcILiDr1JXvtNOygi6OHPMQgU8,1114
+metaflow-stubs/client/core.pyi,sha256=nDpVlelXUrYPvxBFYjLen8o2QUoDrrS4KF3EBYylb5w,47719
+metaflow-stubs/client/filecache.pyi,sha256=cBA2ey1skavJ0vOy6Wv54uEINHIHc677XhzWnwVBP8c,3059
+metaflow-stubs/metadata_provider/__init__.pyi,sha256=UnPA9f9WDQxVdfrnYzV_-vebVX9rzutmGKwEkisBRlg,808
+metaflow-stubs/metadata_provider/heartbeat.pyi,sha256=aE6zjx5msOPdCjuFkMRf1fP49I1uC3Rj0BgN_bX1ZzQ,1068
+metaflow-stubs/metadata_provider/metadata.pyi,sha256=Um9w9YvR88rqrg6AjOAiLBBDrnpsI1LFQvEbUlDVhi4,13235
+metaflow-stubs/metadata_provider/util.pyi,sha256=fgPSzN2btwasrmJFMvd_1L40K15K8hAE_QRPlAFf05Y,758
+metaflow-stubs/mf_extensions/__init__.pyi,sha256=UVwK5WqW7SJr8PpXPZvA7tS-xXqhSGZDjNgRKGbDC8o,516
+metaflow-stubs/mf_extensions/obcheckpoint/__init__.pyi,sha256=sHFOfYqQQ9KbjB8DpmWoktttH0bWtNDdF9Apl4tZ1EM,516
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/__init__.pyi,sha256=Tfb1iDZNTKHq2ApiqwpO8X7en8O0hG7Ac7rVCehKglU,516
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/__init__.pyi,sha256=0SYnSiaAT1WLgPJs65-e3TT2bVSRAoWoVXDeZTszN-c,516
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastructures.pyi,sha256=MYIohECT7GuksDH0Mdn6gFSANy-4N5WVT8IYLPXRnQw,3360
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/exceptions.pyi,sha256=2Y_38KiaSnloGYJbVY8-kcYRVNke6ejU1QqIgdUw3cw,1433
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/__init__.pyi,sha256=WsgPUbslo2NcdLXyw55_Ms9ru_ln8Y8KCk4Q8zeZJ7o,516
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/async_cards.pyi,sha256=8utEinNagLqpMT2yZiyQkT6j0TnwMwxHFLa_fAu6asE,1870
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/deco_injection_mixin.pyi,sha256=ck2IEMFHUWpN20ptEV54C8OP17N0N7RGsP4VH9zJp50,1036
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/extra_components.pyi,sha256=BzQvl6kk9PjgOOhfx5aUM_GGZ5Luj-gga7o0eRL_lkY,3387
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/__init__.pyi,sha256=RKW3XSiESeQt4QamRYtZ2E5T7Bb4d0QDFJ6gLhYE2lQ,516
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/checkpoint_storage.pyi,sha256=m3hsbRgGo3vlBH5Zd6k-9FdOm6ZiJDbRS99MGhReUn8,6417
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/constructors.pyi,sha256=ZlKwZWm4r9gOUx9ZAeIuNnLZluvFk7UN78HZ3JIA_AE,1167
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi,sha256=bxnWu6zHb1u5GY016jju1d4E7kMtSwkGNE6dT6PT8Ek,6324
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi,sha256=xpr9nj0fNXRfWvfkKSDNLPsDZeJ_56bqEL-jz7jddc8,14015
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi,sha256=MmtiGY_AS_oWEQUCKM7umtOtVdBhoNKIW5k6IP3dtK0,988
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi,sha256=5h-cqCHJuXmbrxa7QIKcyUlE5ZNnfqjhg6ESS15N7rI,6797
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/lineage.pyi,sha256=gsF_9cN4PyHLwmCimMDCXbq3hIWc2cglCRcgNy70qH8,1347
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/__init__.pyi,sha256=bBTPlxhHCw-5WrwyOjMmgyTXWsiMuOQqmJyv9fs-U_A,516
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/checkpoint_lister.pyi,sha256=vpTNCnUr1HGgkBqhSaEgACOxEyj3HQKMlm4HuFDdouA,3519
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/lineage_card.pyi,sha256=XOJ9MaWQQl0nkM0Fjk_HBCC7QjbJbDq6YTbnMG613SY,1499
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi,sha256=o41RbNGj_RRWBk9Wuj9oqzn-0zh7H_i7ZTQtGKHkXgw,516
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi,sha256=pLHvswbQarJuToCK-YGv22Viqvktd-6wp832yaJwURE,6422
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi,sha256=hKzpFnXJn3KxOC6fG1l79xHKZ2CowJ9H5F-bkdtUogc,1133
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi,sha256=8eJXIT0Y0Sg7klrj1Z8j7WzTQ2mC4eXagKEb71kn1oE,1026
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/utils.pyi,sha256=Jwcx9iEPdzXuloQVGxJrc7Dc-7PFuUEqSN6Wo_cGKHo,688
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/__init__.pyi,sha256=QXxcrwx7POgqGpRvP8fZZre5MM6b48-Ddicr5mZGNP4,516
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/decorator.pyi,sha256=q-klOHV12jpvBI3OSdT4cqSTj_igz4BYSqH7vas8tMs,9385
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/__init__.pyi,sha256=jzIxF6n_hr7LrrQWSia3mNE4OjOa3N4t3qQccfVJk7c,516
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/core.pyi,sha256=1Lr7i0Od0xsSucggcPkWAf0UQEjCsXTV_4uNfaauNRY,4492
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/exceptions.pyi,sha256=CN_DLSUGbXzjcwUxDBZvsGarHSdrI4jdS6ZRlkB7py4,968
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/model_storage.pyi,sha256=V2XGI4YRoeBdGCo7QHBHKuJGOPro3jupBpyjARVLV_I,3449
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/__init__.pyi,sha256=cpTFCyO_Quw3sb3TrWWCxQQ6nlSoAx-A-XGrQCydTUQ,516
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/flowspec_utils.pyi,sha256=XYuJFnIHdsJjeD-oQ51FblTTLDbEEC3uksjjpvvUhC0,2481
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/general.pyi,sha256=of5pXAjeO_Mx6pbwRTl7JomlCEg1cWxFpGvfROAbRVY,1694
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/identity_utils.pyi,sha256=7TMGD8eDs2-iMdx3zos2n7a99rSb5-E0LYwbdn0YB0s,2135
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi,sha256=PUVd-aowa6w8U6bcu1FPHG_8jux84_D4n79jktMgXEo,1887
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/__init__.pyi,sha256=hB-j_3OjmdEGnLfXL_t368L_s7-VmUaz2dGVKVF_oog,516
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/base.pyi,sha256=qrRp-Ee7DvEPKRmFLarebrgSJD4HMowO1xeEIuycAxs,823
+metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/tar.pyi,sha256=DLjIX3LNLaT2T2RlVaPFL7eUI1PPXmO8trKKOjQk4ds,1383
+metaflow-stubs/mf_extensions/outerbounds/__init__.pyi,sha256=GTS8ioC3Xz34U-TDsNn-FHNhkdDlslwjc3QT0dPwzKM,516
+metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi,sha256=0wauTg1nTsK6egroNcVbXKhy-6iwQTrcrYA_ykE8edI,1642
+metaflow-stubs/mf_extensions/outerbounds/plugins/__init__.pyi,sha256=GNude0bt_DpZRlS7bubxVUFrOfWNsQQU8Bc55SGRIsY,516
+metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/__init__.pyi,sha256=nR8MlGWqI0vzoeK35acsQfF4UK7TA_lU_yHhQfLazQo,516
+metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/injector.pyi,sha256=mXXoSe27Y5xibUJ7fYUtRIN50Bv3PMY4GyyIjRCraxg,1035
+metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/__init__.pyi,sha256=I53It4UDrILFgJMIAOO3aztNfr92DIa67jd70ZmXJfo,516
+metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/ollama.pyi,sha256=gredU6QOo9PkrFGY5efuazA571r-NBLtcpveyth7E4M,1112
+metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi,sha256=mLB_1sZhEeejXyTQ46r8IogrdB9K__d1RQL5GXB8J3w,516
+metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi,sha256=6LuLYahZEnPT6gjXmTrYHVTObjHY7wbH2cUxDLErK2c,3281
+metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi,sha256=m1ujIKXfNyoz1litZaZR4WKO8pfHDJA50LLG-_CR1NM,516
+metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi,sha256=kF4zCPwRW4DJwOEQeYpmKi8E_aKlGXuCKABFq0S8KCc,4478
+metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi,sha256=JAYkZxk13qSH2XLOut4HgnCnbDKI-k5ntO9MSxGBLRc,516
+metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi,sha256=egwd75E5P5za2AQd-nAVElGRUH9WeRqPCZKsFCGSnN4,810
+metaflow-stubs/plugins/__init__.pyi,sha256=Ve2DlEcpWteSGYs2Xd7Gd2fvYHAetsTfNbKqtWriKxY,3807
+metaflow-stubs/plugins/catch_decorator.pyi,sha256=Q5Y-EsfwSL9ZAWPHcG8S-keV37qBZRmyFzBJtAbolY0,2243
+metaflow-stubs/plugins/debug_logger.pyi,sha256=j_RhQB-OtgNjxC_0DlsrqqOI8Wchm8U9vyX2mZzu6vs,932
+metaflow-stubs/plugins/debug_monitor.pyi,sha256=liV1lnKneDq04JMzbssoF0Vd1ZHUuGMuKufXZyWdWuw,910
+metaflow-stubs/plugins/environment_decorator.pyi,sha256=gfqHIjjFDqRhuWagoVkpB7q4wiFzXUa7qr0F0XAmFeQ,1052
+metaflow-stubs/plugins/events_decorator.pyi,sha256=UedFdE4IwXJw2bXIeCM6HT5Fd1eU_GGTWR8tzWk2LV0,4772
+metaflow-stubs/plugins/parallel_decorator.pyi,sha256=O9l_lm0kO2RPZz5tbIvjGde-d0IBCgZc0YKJDrpSfY4,2445
+metaflow-stubs/plugins/perimeters.pyi,sha256=6clKg96uQOcR3audT0Lyt9pFybg-aeyKWlwvKx1pOKM,811
+metaflow-stubs/plugins/project_decorator.pyi,sha256=hIiJ49c71hyt4X4UkDBI6lrSXqyHPalDfPfDFI27hf8,3713
+metaflow-stubs/plugins/resources_decorator.pyi,sha256=OYlZ3m1L6V7ayqAGzTuOx1wXcN-FqruSZLyNa4OvVZQ,1829
+metaflow-stubs/plugins/retry_decorator.pyi,sha256=XF8Ny-0qicE6tZbYnILySlJ71pJhrspU-gqq3TBCYH8,1709
+metaflow-stubs/plugins/storage_executor.pyi,sha256=M3myCGrLb8l8t-bBrwd1qTPFthVeZ8UR2Qxd6DeZMBg,1198
+metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi,sha256=tv-3Frf7wWdUX-dnKBTiboFxjW0Mvc7o5EUqVKHoUno,2087
+metaflow-stubs/plugins/timeout_decorator.pyi,sha256=Q8-AgQnQ2x4SKp8OksNQlInhHNiUpLY7vcLKp9LZtQE,2296
+metaflow-stubs/plugins/airflow/__init__.pyi,sha256=QgGXkKMhoa6d-8bUmuFOHW20OoEKeEOi6CaoINmhX9c,669
+metaflow-stubs/plugins/airflow/airflow_utils.pyi,sha256=csMX6ou_3ZE71IRiUomJwas7p38ukY8YmxFfwMtrNfE,3154
+metaflow-stubs/plugins/airflow/exception.pyi,sha256=2VZvJnxqX9jNpQKWqZLamasT42niXH6Nqk1iM2Kkms4,923
+metaflow-stubs/plugins/airflow/sensors/__init__.pyi,sha256=Uwic6dqVfmUAvKBKjXTJ3Mu5oNp7H6nVnHoa9zDbMGg,877
+metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi,sha256=azdAd0gMyb-ehRMETaVrspS2Nx-SEDo6NUfLD60d5P0,1497
+metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi,sha256=o11rnIGE51ijp0jZ-s8pY3NBXdVSUl8sfBbHrYOZ0vM,3309
+metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi,sha256=ZqfCj-oDDkVj0wLv0icp5pHfnxQJ2E9nIMtFV4qfPIw,3220
+metaflow-stubs/plugins/argo/__init__.pyi,sha256=Vz1hX3MwOG2jyWgtkFzVUoo9SLgSnBDP3k0MXb-q63c,727
+metaflow-stubs/plugins/argo/argo_client.pyi,sha256=H6m3gwkiHk9UVLd5-Y4NtOXwjKn-3EHHLIWu8BClCik,2514
+metaflow-stubs/plugins/argo/argo_events.pyi,sha256=VLOpkbdR6fJd-LuBJEXQ5Onw4uSOsZke31tplI7t51s,3348
+metaflow-stubs/plugins/argo/argo_workflows.pyi,sha256=LYirMMxR2IOYszCXgjnmA0wJVIh_NvxgB5HXNQ7_EJQ,13674
+metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi,sha256=1lVnZN4mfsy-9qa4opMAk5U0r4p5OMAEdkISU7lBtcY,1375
+metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi,sha256=-cAYTB7nEh3BN0V9S_RHfv95dDEtS7tjjjos6Jfhzi0,4595
+metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi,sha256=zwbeAGQevREwYkDL_mXwhdwiNpHDjn6yK22zewDOGZo,6061
+metaflow-stubs/plugins/aws/__init__.pyi,sha256=mzmtRLplZrim82NlemZmP918lmcXHMUYlx8U8-RbtDg,755
+metaflow-stubs/plugins/aws/aws_client.pyi,sha256=a2MBksrYH8bkANAYvgMYcgbB0JWuWQ_eFrAPUDaPjqc,930
+metaflow-stubs/plugins/aws/aws_utils.pyi,sha256=T04eyxiPxoVCEI2xUqdx7VlcWq-nEuby1XFfC-9fE44,3147
+metaflow-stubs/plugins/aws/batch/__init__.pyi,sha256=EfRevcY_auJCzuCi9Nz6cKneWVTk3zCEtVkX6KG_wH4,675
+metaflow-stubs/plugins/aws/batch/batch.pyi,sha256=0kXE4xmPJPgKmaDOurROoY6KYzYrlnQ09Y6jOEJdFLk,2905
+metaflow-stubs/plugins/aws/batch/batch_client.pyi,sha256=-57wh0eUEqocfqSY4YsDkn8t_BuHLD7l-94CM7163kc,3767
+metaflow-stubs/plugins/aws/batch/batch_decorator.pyi,sha256=r3-aa5JC5qcwkJv7uNG5zLStqcAXr1ZGGUD1WnVLcs0,5937
+metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi,sha256=S3OGnSSyWQ-LBsSC3X8IUC4whf4F0bWIenOb_cdrY9A,645
+metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi,sha256=-_KGSlyRZsZroFyNY1lAhjKd0b5NTLEmP1aNZwOWNrc,3575
+metaflow-stubs/plugins/aws/step_functions/__init__.pyi,sha256=cve3DIqLROwY13Shu4xzQigO0L8HdnUnW70zyo9BhE0,674
+metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi,sha256=z0il5wlSoYknR3q5Zyg9NdyqFizvAufgVy76FrXshFw,899
+metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi,sha256=npTBZlzrRjuX0wxadHRYP-P8Wd0kFAI03w69TrmSHWg,1637
+metaflow-stubs/plugins/aws/step_functions/step_functions.pyi,sha256=iV6ZZf44WEPRGXcu41EX0_9jPysFYAaNPyj8rJVzgQI,4553
+metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi,sha256=iUU-MilehEHV_pyczZW5DF3MEWSObjYV12dPnsHa5UM,1186
+metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi,sha256=1Ux9gA4P5sSjZQK1J0wmoAbk5ZwO3nymU5E_p3nkJQU,3881
+metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi,sha256=SbAUeHcfw4Xa8rTblWiT3GtTQpFIn2Q3qJQ_aWizRMQ,4342
+metaflow-stubs/plugins/azure/__init__.pyi,sha256=KcwTNuiALU-dKn_fzaou-dOT0lhNtea47sr9XPpT_2A,1011
+metaflow-stubs/plugins/azure/azure_credential.pyi,sha256=HRJ263MVQCWyOTjHKdJEiqmOIzgVx69Rmg_xjQbKq8I,1168
+metaflow-stubs/plugins/azure/azure_exceptions.pyi,sha256=AZMV2VJIC4BAQzccOThx8ZNx41UJb2TLH54l-1Yl7UI,999
+metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi,sha256=x6Q2GwIrUgFMY7drvhHU4F000rNNwntGfojNi5wsYEA,2289
+metaflow-stubs/plugins/azure/azure_utils.pyi,sha256=xckmCYWkHeka-r7IUJhzOm78AkcFOyRuh6YQSeyBPbo,2458
+metaflow-stubs/plugins/azure/blob_service_client_factory.pyi,sha256=ds3xFlZs74XH_lllBL7iKGb7XfGlG_2FGAEAmRCXSyU,1568
+metaflow-stubs/plugins/azure/includefile_support.pyi,sha256=q_CzVIMe5whtVFWNq18QotgMfXGWOAl4574xRentv9U,1667
+metaflow-stubs/plugins/cards/__init__.pyi,sha256=LSzS4KRVR0nnhLFt6gy9tLpyv7hG44oO0458jnMtWwc,916
+metaflow-stubs/plugins/cards/card_client.pyi,sha256=YJaGXQr2yUejyG7uCCa_wGc0b3D_0YH2BKfhEz5c5e8,4909
+metaflow-stubs/plugins/cards/card_creator.pyi,sha256=A_SrnJOvIUbU1Smpy5HOXLFSKqDeHv9JeE4e3s5XR3E,1089
+metaflow-stubs/plugins/cards/card_datastore.pyi,sha256=GAk7h13vKMcLdTWk_FfANU8sb2QYLNBjqA8AN-PL2i4,2962
+metaflow-stubs/plugins/cards/card_decorator.pyi,sha256=0nJbvT0TKUCwiJQIyiWkwF8jlD5EZSGPq4iaqSlhcts,3038
+metaflow-stubs/plugins/cards/card_resolver.pyi,sha256=MT1L7Fq6X14fZhIpgv_uQdJzsxCs4I_MHJJsHDY4xLM,759
+metaflow-stubs/plugins/cards/component_serializer.pyi,sha256=VOqBf24WnVJzoMzMR_FBXSxuJ5g5rTTBUgKCiYDeyPU,8476
+metaflow-stubs/plugins/cards/exception.pyi,sha256=nryZmKcjFUHsXzH375dKbVCnFAnSINeN4nJzXOJASuo,2470
+metaflow-stubs/plugins/cards/card_modules/__init__.pyi,sha256=b0hdXbB8xFQdixb6vFoIANIWoqGzH7y6NQCMUZh5CsE,986
+metaflow-stubs/plugins/cards/card_modules/basic.pyi,sha256=bEZSyVtU8_-DLTC4e6l4TU-fmVJtRv5l_GuGCP7OK6o,7502
+metaflow-stubs/plugins/cards/card_modules/card.pyi,sha256=Wig_Y-mAXDZX77R7UzYyfq9KcrAQSiYAIk92e2TlSm4,3196
+metaflow-stubs/plugins/cards/card_modules/components.pyi,sha256=lcjEdHFfjob_esutWwCJgXKgb_KA2iw-McY5Yvd8_oI,11481
+metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi,sha256=IO0ALr4KG6hcoUaskBLeptIt4vo-6dcWOj6i8SC3GAw,1479
+metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi,sha256=xqjB51mUbQxiSIOY4E5kg5qb2AQFuiqaRTvbCa5NVTI,853
+metaflow-stubs/plugins/cards/card_modules/test_cards.pyi,sha256=eK9j6i4LphJs3s5AmM_M3lT2PQWpntnagbNFD6Quy24,3550
+metaflow-stubs/plugins/datatools/__init__.pyi,sha256=cjbEGoGPvzTXYyYldRDuhY5pCFGSU7AWmUPW2He83xI,933
+metaflow-stubs/plugins/datatools/local.pyi,sha256=z5TWVEro5RuF5VGLcVbZgO3wqnyZl4sJucsafcx-DYA,2797
+metaflow-stubs/plugins/datatools/s3/__init__.pyi,sha256=ZXMD5R8RHBEtnmG-lO5MRtmh2m8FWJL6-FVGfY9RMZY,1204
+metaflow-stubs/plugins/datatools/s3/s3.pyi,sha256=ilelAgNUVUFW6E1ChaB2aJJyS59LK32ohgmK8T7nwBQ,23702
+metaflow-stubs/plugins/datatools/s3/s3tail.pyi,sha256=2sKwzp9EMgBNeDJukFAwvTYb2zzKc8dyNYs41Pe2qBE,1036
+metaflow-stubs/plugins/datatools/s3/s3util.pyi,sha256=4rvRsRFIPmNjVchmdcE3chGJkDOQ7215FMAFsBrFLyI,1132
+metaflow-stubs/plugins/frameworks/__init__.pyi,sha256=yL16AH-pNrVm7rCCF5YRfDF8TIVtiEN_qv01L3JKRPU,587
+metaflow-stubs/plugins/frameworks/pytorch.pyi,sha256=J3lXs4c7APEukHMWgwld4LCrHNz1DXI6yuvbLIxfUho,1200
+metaflow-stubs/plugins/gcp/__init__.pyi,sha256=qlBRvPgWZmiwbHGlWat4tKAOK8kGhbOnuTclr1MGlvY,923
+metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi,sha256=NLzOX6MrilJMiJVmnUInYDG7AAn8083uBiBsTLpyMKU,3298
+metaflow-stubs/plugins/gcp/gs_exceptions.pyi,sha256=IS38w86IhM6n8XGispRCkVKxscdh2QYXDZtkmGvEBfA,796
+metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi,sha256=8mL6Jghtn-kkQB7AlUuI3ERAlLxMOiIayMHD0qvkQWg,899
+metaflow-stubs/plugins/gcp/gs_utils.pyi,sha256=AsHa-24b02tYoIuObSW1g6IN77LQbweNZUiqQdNF1Dw,1086
+metaflow-stubs/plugins/gcp/includefile_support.pyi,sha256=pKTvtGte6jH2VWd-ZkdvGS7pXwLcgjgor_cvnYNW7vA,1641
+metaflow-stubs/plugins/kubernetes/__init__.pyi,sha256=ZoYIa_TxPVEXSUYgN4LkL3e36wCRZMmxuG8-b1vGqGs,803
+metaflow-stubs/plugins/kubernetes/kube_utils.pyi,sha256=R55LFQ-iqft-Qfbbd6tZKmvYtVXlHGvm8WtEd5u5B_c,1707
+metaflow-stubs/plugins/kubernetes/kubernetes.pyi,sha256=ifzK1ycWJB_rheiPAse3vXSatBnag5HSrBJX1TuKETA,3765
+metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi,sha256=EHNGc2ndw3Z65Q_2noepgozzCzE6EmDfm47iPO45vzI,1198
+metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi,sha256=F_7ud3Yt8v1LdL3uJ4tB8xCgHUr2QUyrqkDDko0aqr8,7205
+metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi,sha256=mTbUKyupkiyzAmkX-ZIWqDjACHrAuEFcf5673N5Q4es,5389
+metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi,sha256=qyXpzyGkoSasFzGuJlJOKuBgDFRwiV8Upid1ClZoBYc,762
+metaflow-stubs/plugins/ollama/__init__.pyi,sha256=4nhJTwx055YAhx1ygOE7YJo_NLvBX76_bCW4VUeWMhQ,2199
+metaflow-stubs/plugins/pypi/__init__.pyi,sha256=3r96dJ5s7dGNFvDjHwI4zwrNv9tEIrCw522CPlFyIbE,915
+metaflow-stubs/plugins/pypi/conda_decorator.pyi,sha256=n_6WTuJz92kvZctZVmFFz2MrUzKmny12rpToSSG5NAs,3789
+metaflow-stubs/plugins/pypi/conda_environment.pyi,sha256=NdJ4ITM2nnDcGoBneQe33KSIawVjlZMZFThaiuKGJD4,2409
+metaflow-stubs/plugins/pypi/pypi_decorator.pyi,sha256=QZfosmZgVxuM1Mvjwnmks-ZtMPJ3OTYBwGDLQmQm6L0,2610
+metaflow-stubs/plugins/pypi/pypi_environment.pyi,sha256=xKha0NukB6Y4W9c7jZ91blSFt6fldyiufds2B8HL-1s,834
+metaflow-stubs/plugins/pypi/utils.pyi,sha256=JxIcq2nDu-WcXr6bp-8r06EtZEgNsGRoYINT45WsjmQ,874
+metaflow-stubs/plugins/secrets/__init__.pyi,sha256=buqyZT1O_14lYPRqYaaVQg0WrSwhNFU1VI11ES2Z5Gc,1042
+metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi,sha256=xcYIuent3rmq0xT8FdUg6PqCv8HPSChChzKC-5aol9o,986
+metaflow-stubs/plugins/secrets/secrets_decorator.pyi,sha256=Th7ZKVxWWf27Qbb7kXJn-DXrYShQ0NCDec-M9mHpz6Q,2345
+metaflow-stubs/plugins/snowflake/__init__.pyi,sha256=lFCD-gOS7qILABldlnwhNK2iS8vgSs3Jv-OiAhRF8cU,928
+metaflow-stubs/profilers/__init__.pyi,sha256=WN3YNxcdqqpp7JAtLTIyZfJ91keBnkBwY_8hTweuO9Q,696
+metaflow-stubs/runner/__init__.pyi,sha256=5US5UeDXxCxrQ57EN_dqefCFRD4YiGF_d6qZwM9J62Y,831
+metaflow-stubs/runner/deployer.pyi,sha256=_BJVP-X2Wy6f8Dcq3k0ewX94RvoC-dgbKV_Rn5KvSHs,7538
+metaflow-stubs/runner/deployer_impl.pyi,sha256=AA_soLHhiWFp0HPqUIqsSaRvxboX6fBH10BX2Pz53Ek,3313
+metaflow-stubs/runner/metaflow_runner.pyi,sha256=sq3t_vGQyIrbvNHwysbbH6v6AolVq2kVa6Swou6S_Fc,10893
+metaflow-stubs/runner/nbdeploy.pyi,sha256=HupMw0PiQOERracG_mP4Mx7133egqdeiFoMxjULbvnY,2946
+metaflow-stubs/runner/nbrun.pyi,sha256=VJ3sjfwRNN_9x9Ubbwmd8y_azCjGdOQAaWyj37y4UEQ,6035
+metaflow-stubs/runner/subprocess_manager.pyi,sha256=WxDxJXDY7McuvpIwjeQRk42eljf2VpTVRxXoOYL2TCY,9273
+metaflow-stubs/runner/utils.pyi,sha256=-2QSOVgJI0LSfyPoRLHTCWcZ7gdN-mUo-MXs5yMtSjI,5936
+metaflow-stubs/system/__init__.pyi,sha256=iBgz7hvfrDw3LmdSYBsmQfsuDWlpXJdIntTvAEw4usY,761
+metaflow-stubs/system/system_logger.pyi,sha256=LaIBAoDVt1Lkp9GwJkxRDjfqoD_dICak29RAOq0w8lE,1666
+metaflow-stubs/system/system_monitor.pyi,sha256=uobyhd3Os5QfteL75nayZoSTqhADeTSqQhFyxoYtfXU,1788
+metaflow-stubs/user_configs/__init__.pyi,sha256=kXRi_G-D-51XrAK3G160KPkwFyvKOnXqtLbeWMqQl74,707
+metaflow-stubs/user_configs/config_decorators.pyi,sha256=NJQJ0At5e-Dm8z2QJ9iVj1ii7AL8YwwLEHbkMBJWocc,8940
+metaflow-stubs/user_configs/config_options.pyi,sha256=kY_oui_SIQ_BcKMUSbidcComJcLuiHB9wI7Unu_woSU,2955
+metaflow-stubs/user_configs/config_parameters.pyi,sha256=teIGUk1eGe9QGzf32GJiBe2tw35C0zh18u8UWT8O-zE,8349
+ob_metaflow_stubs-6.0.3.144.dist-info/METADATA,sha256=I_MvrEoTcWfoAtRiCkt30ymdUBXmUI9qw7IKURh2dVw,657
+ob_metaflow_stubs-6.0.3.144.dist-info/WHEEL,sha256=Kh9pAotZVRFj97E15yTA4iADqXdQfIVTHcNaZTjxeGM,110
+ob_metaflow_stubs-6.0.3.144.dist-info/top_level.txt,sha256=vEJN-gFv184nGYaYbnS_0-Q3cmXuRw3iT9l3yyYb1JA,15
+ob_metaflow_stubs-6.0.3.144.dist-info/RECORD,,

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/context.pyi

@@ -1,79 +0,0 @@
-######################################################################################################
-#                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.051021                                                            #
-######################################################################################################
-
-from __future__ import annotations
-
-import metaflow
-import typing
-if typing.TYPE_CHECKING:
-    import metaflow.exception
-    import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.context
-    import metaflow
-
-from ......exception import MetaflowException as MetaflowException
-
-TYPE_CHECKING: bool
-
-ARTIFACT_STORE_CONFIG_ENV_VAR: str
-
-class UnresolvableDatastoreException(metaflow.exception.MetaflowException, metaclass=type):
-    ...
-
-class ArtifactStoreContext(object, metaclass=type):
-    """
-    This class will act as a singleton to switch the datastore so that
-    Any checkpoint operations will be done in the correct datastore. This
-    can be a useful way of short-circuting a datastore switch between runtime
-    And post-runtime retrieval operations.
-    """
-    @property
-    def MF_DATASTORES(self):
-        ...
-    def __init__(self):
-        ...
-    def flow_init_context(self, context):
-        ...
-    def post_task_runtime_context(self, context):
-        ...
-    def switch_context(self, context):
-        ...
-    def get(self):
-        ...
-    def from_task_metadata(self, task: "metaflow.Task"):
-        """
-        TODO: This is not the best way to do this since this is mostly used for
-        listing checkpoints post-runtime. It is also currently a code path for
-        `Checkpoint.list()` but it can be clunky and I havent thought throught its
-        implications on other code paths. Currently it can work with :
-        ```
-        config = {
-            "client_params":{
-                "access_key_id": "abc",
-                "secret_access_key": "xyz"
-            }
-        }
-        with using_artifact_store(type="s3", config=config):
-            Checkpoint().load(Checkpoint().list(task=task)[0], "my-path")
-        ```
-        """
-        ...
-    def to_task_metadata(self):
-        ...
-    def default(self):
-        ...
-    ...
-
-datastore_context: ArtifactStoreContext
-
-def using_artifact_store(type, config):
-    """
-    This context manager can be used to switch the artifact store
-    for a block of code. This is useful when users maybe accessing
-    checkpoints/models from a different datastore using the
-    `@with_artifact_store` decorator.
-    """
-    ...
-

metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/decorator.pyi

@@ -1,46 +0,0 @@
-######################################################################################################
-#                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.14.0.2rc0+obcheckpoint(0.1.8.1rc0);ob(v1)                                            #
-# Generated on 2025-02-17T19:27:36.102835                                                            #
-######################################################################################################
-
-from __future__ import annotations
-
-import metaflow
-import typing
-if typing.TYPE_CHECKING:
-    import metaflow.decorators
-
-from ......metadata_provider.metadata import MetaDatum as MetaDatum
-from ......exception import MetaflowException as MetaflowException
-
-class ArtifactStoreFlowDecorator(metaflow.decorators.FlowDecorator, metaclass=type):
-    """
-    Allows setting external datastores to save data for the
-    `@checkpoint`/`@model`/`@huggingface_hub` decorators. This is useful
-    when the compute medium lives in a different geographical location
-    than metaflow's datastore (e.g. S3, GCS, Azure Blob Storage).
-    
-    Parameters:
-    ----------
-    
-    type: str
-        The type of the datastore. Can be one of 's3', 'gcs', 'azure' or any other supported metaflow Datastore.
-    
-    config: dict or Callable
-        Dictionary of configuration options for the datastore. The following keys are required:
-        - root: The root path in the datastore where the data will be saved. (needs to be in the format expected by the datastore)
-            - example: 's3://bucket-name/path/to/root'
-            - example: 'gs://bucket-name/path/to/root'
-            - example: 'https://myblockacc.blob.core.windows.net/metaflow/'
-        - role_arn (optional): AWS IAM role to access s3 bucket (only when `type` is 's3')
-        - session_vars (optional): AWS session variables to access s3 bucket (only when `type` is 's3')
-        - client_params (optional): AWS client parameters to access s3 bucket (only when `type` is 's3')
-    """
-    def flow_init(self, flow, graph, environment, flow_datastore, metadata, logger, echo, options):
-        ...
-    ...
-
-def set_datastore_context(flow, metadata, run_id, step_name, task_id, retry_count):
-    ...
-