dagstermill 0.23.9__py3-none-any.whl → 0.23.10__py3-none-any.whl

dagstermill/asset_factory.py

@@ -21,6 +21,7 @@ from dagster._config.pythonic_config.type_check_utils import safe_is_subclass
 from dagster._core.definitions.events import CoercibleToAssetKey, CoercibleToAssetKeyPrefix
 from dagster._core.definitions.utils import normalize_tags
 from dagster._core.execution.context.compute import OpExecutionContext
+from dagster._core.storage.tags import COMPUTE_KIND_TAG
 
 from dagstermill.factory import _clean_path_for_windows, execute_notebook
 
@@ -85,6 +86,7 @@ def define_dagstermill_asset(
     retry_policy: Optional[RetryPolicy] = None,
     save_notebook_on_failure: bool = False,
     non_argument_deps: Optional[Union[Set[AssetKey], Set[str]]] = None,
+    asset_tags: Optional[Mapping[str, Any]] = None,
 ) -> AssetsDefinition:
     """Creates a Dagster asset for a Jupyter notebook.
 
@@ -123,6 +125,7 @@ def define_dagstermill_asset(
         save_notebook_on_failure (bool): If True and the notebook fails during execution, the failed notebook will be
             written to the Dagster storage directory. The location of the file will be printed in the Dagster logs.
             Defaults to False.
+        asset_tags (Optional[Dict[str, Any]]): A dictionary of tags to apply to the asset.
         non_argument_deps (Optional[Union[Set[AssetKey], Set[str]]]): Deprecated, use deps instead. Set of asset keys that are
             upstream dependencies, but do not pass an input to the asset.
 
@@ -180,12 +183,15 @@ def define_dagstermill_asset(
             " is reserved for use by Dagster",
         )
         check.invariant(
-            "kind" not in op_tags,
-            "user-defined op tags contains the `kind` key, but the `kind` key is reserved for"
+            COMPUTE_KIND_TAG not in op_tags,
+            f"user-defined op tags contains the `{COMPUTE_KIND_TAG}` key, but the `{COMPUTE_KIND_TAG}` key is reserved for"
             " use by Dagster",
         )
 
-    default_tags = {"notebook_path": _clean_path_for_windows(notebook_path), "kind": "ipynb"}
+    default_tags = {
+        "notebook_path": _clean_path_for_windows(notebook_path),
+        COMPUTE_KIND_TAG: "ipynb",
+    }
 
     if safe_is_subclass(config_schema, Config):
         config_schema = infer_schema_from_config_class(cast(Type[Config], config_schema))
@@ -207,6 +213,7 @@ def define_dagstermill_asset(
         io_manager_key=io_mgr_key,
         retry_policy=retry_policy,
         non_argument_deps=non_argument_deps,
+        tags=asset_tags,
     )(
         _make_dagstermill_asset_compute_fn(
             name=name,

dagstermill/examples/repository.py

@@ -570,6 +570,7 @@ custom_io_mgr_key_asset = dagstermill.define_dagstermill_asset(
     name="custom_io_mgr_key",
     notebook_path=nb_test_path("hello_world"),
     io_manager_key="my_custom_io_manager",
+    asset_tags={"foo": "bar"},
 )
 
 yield_event_asset = dagstermill.define_dagstermill_asset(

dagstermill/factory.py

@@ -26,6 +26,7 @@ from dagster._core.execution.context.compute import OpExecutionContext
 from dagster._core.execution.context.input import build_input_context
 from dagster._core.execution.context.system import StepExecutionContext
 from dagster._core.execution.plan.outputs import StepOutputHandle
+from dagster._core.storage.tags import COMPUTE_KIND_TAG
 from dagster._serdes import pack_value
 from dagster._seven import get_system_temp_directory
 from dagster._utils import mkdir_p, safe_tempfile_path
@@ -420,11 +421,14 @@ def define_dagstermill_op(
             " is reserved for use by Dagster",
         )
         check.invariant(
-            "kind" not in tags,
+            COMPUTE_KIND_TAG not in tags,
             "user-defined op tags contains the `kind` key, but the `kind` key is reserved for"
             " use by Dagster",
         )
-    default_tags = {"notebook_path": _clean_path_for_windows(notebook_path), "kind": "ipynb"}
+    default_tags = {
+        "notebook_path": _clean_path_for_windows(notebook_path),
+        COMPUTE_KIND_TAG: "ipynb",
+    }
 
     if safe_is_subclass(config_schema, Config):
         config_schema = infer_schema_from_config_class(cast(Type[Config], config_schema))

dagstermill/version.py

@@ -1 +1 @@
-__version__ = "0.23.9"
+__version__ = "0.23.10"

{dagstermill-0.23.9.dist-info → dagstermill-0.23.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagstermill
-Version: 0.23.9
+Version: 0.23.10
 Summary: run notebooks using the Dagster tools
 Author: Dagster Labs
 Author-email: hello@dagsterlabs.com
@@ -13,7 +13,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.8,<3.13
 License-File: LICENSE
-Requires-Dist: dagster ==1.7.9
+Requires-Dist: dagster ==1.7.10
 Requires-Dist: ipykernel !=5.4.0,!=5.4.1,>=4.9.0
 Requires-Dist: ipython-genutils >=0.2.0
 Requires-Dist: packaging >=20.9

{dagstermill-0.23.9.dist-info → dagstermill-0.23.10.dist-info}/RECORD

@@ -1,22 +1,22 @@
 dagstermill/__init__.py,sha256=WZRYSjL1yAM1w5cqmC1t2T7hvuW-RGGY5tV9vZGHZwU,1133
 dagstermill/__main__.py,sha256=bYt9eEaoRQWdejEHFD8REx9jxVEdZptECFsV7F49Ink,30
-dagstermill/asset_factory.py,sha256=3moIwVr3dzFuDRVAdRZlQlO00nfGFAiA8cvmOgJi1zw,9463
+dagstermill/asset_factory.py,sha256=_or3ENKhrxV2EDQjpbKbv6Sf28aehbthUKEu9VxO53U,9759
 dagstermill/cli.py,sha256=lffeaQgenSR7s8MhNUBNA_BeVehlMUrdg7gnQl1WTdY,4494
 dagstermill/compat.py,sha256=GCfUaGC3eIEhlZP_VFtua8hFKNtwXocOAfYxuKZ5X3I,526
 dagstermill/context.py,sha256=3_SkRBM9wDB6T64Lc4yJZGp8mpA_PMXkHlF8U1I19A4,6104
 dagstermill/engine.py,sha256=4CZW-eni4TnG8C8VhPkZZgj6djItdOUKhHPzvaVhaYo,5843
 dagstermill/errors.py,sha256=WOmpAGp-J1XhyGK_LT3ZZKsBwF5dvrWbqSaSldtoh6Y,141
-dagstermill/factory.py,sha256=R9IiuAFi35nHneuWsegcXEH1iTp4qUuO0zmtizmBZb0,18638
+dagstermill/factory.py,sha256=mfyw85jFGHj60_tx08jRsiauJnVifqgexk1n-Op6760,18737
 dagstermill/io_managers.py,sha256=D9UA4KufzrOeCan3TJyPogD0uIJBjK80ExikdMHqweI,4403
 dagstermill/manager.py,sha256=HPk1PtVOc7MDFb3F3t-XECvhPYRPRAXNBSapXYAIUBA,15788
 dagstermill/serialize.py,sha256=eXW3c26CiILT_uebyFcAKBnsiNxnjyGT_ch3PfGyjek,188
 dagstermill/translator.py,sha256=h1VPAOWtdjLKzFjRmyN9hO_R6qJuAETNkJydfdgwWGM,2170
-dagstermill/version.py,sha256=ooBNYu742lMnRkf6FXz0qW9XX29Ku8ZtCGCEiI8IjMI,23
+dagstermill/version.py,sha256=MubiA1vMagp3LPGUXEemPTyr2p4sR0ljgXGsAb2yT-Q,24
 dagstermill/examples/__init__.py,sha256=kzan-9zFjxaJ8o9bqUso44gcGiOmJrlq4JYO-yIBQao,55
-dagstermill/examples/repository.py,sha256=hKQ1_GzHFiSf4oZmr1HYQmyRSMG0HDq25m0Y0oz0eXQ,15827
-dagstermill-0.23.9.dist-info/LICENSE,sha256=TMatHW4_G9ldRdodEAp-l2Xa2WvsdeOh60E3v1R2jis,11349
-dagstermill-0.23.9.dist-info/METADATA,sha256=Q0RsNrFGsD0oHodzPcE5d4UH58x7JXaqpfU3F4UP1wo,985
-dagstermill-0.23.9.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-dagstermill-0.23.9.dist-info/entry_points.txt,sha256=885a7vvhABYWEj7W28elkzSmIcKO3REkdd5h4Z4DEJs,53
-dagstermill-0.23.9.dist-info/top_level.txt,sha256=YDelJKdA5YIIrjsObdd8U4E9YhuXJLRe9NKfUzud9Uc,12
-dagstermill-0.23.9.dist-info/RECORD,,
+dagstermill/examples/repository.py,sha256=AlBcvZHCqZ4wAGp8YjO67A1STx0FPhBG97vGEjZv3Bw,15858
+dagstermill-0.23.10.dist-info/LICENSE,sha256=TMatHW4_G9ldRdodEAp-l2Xa2WvsdeOh60E3v1R2jis,11349
+dagstermill-0.23.10.dist-info/METADATA,sha256=9QhGK5g0PUeto9rGFPDDb0VZB-zQ5lLV2-po3-6nCX0,987
+dagstermill-0.23.10.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+dagstermill-0.23.10.dist-info/entry_points.txt,sha256=885a7vvhABYWEj7W28elkzSmIcKO3REkdd5h4Z4DEJs,53
+dagstermill-0.23.10.dist-info/top_level.txt,sha256=YDelJKdA5YIIrjsObdd8U4E9YhuXJLRe9NKfUzud9Uc,12
+dagstermill-0.23.10.dist-info/RECORD,,