dagstermill 0.25.12__py3-none-any.whl → 0.26.0__py3-none-any.whl

dagstermill/asset_factory.py

@@ -17,6 +17,7 @@ from dagster import (
     SourceAsset,
     asset,
 )
+from dagster._annotations import beta, beta_param
 from dagster._config.pythonic_config import Config, infer_schema_from_config_class
 from dagster._config.pythonic_config.type_check_utils import safe_is_subclass
 from dagster._core.definitions.events import CoercibleToAssetKey, CoercibleToAssetKeyPrefix
@@ -69,6 +70,8 @@ def _make_dagstermill_asset_compute_fn(
     return _t_fn
 
 
+@beta
+@beta_param(param="resource_defs")
 def define_dagstermill_asset(
     name: str,
     notebook_path: str,
@@ -117,7 +120,7 @@ def define_dagstermill_asset(
         group_name (Optional[str]): A string name used to organize multiple assets into groups. If not provided,
             the name "default" is used.
         resource_defs (Optional[Mapping[str, ResourceDefinition]]):
-            (Experimental) A mapping of resource keys to resource definitions. These resources
+            (Beta) A mapping of resource keys to resource definitions. These resources
             will be initialized during execution, and can be accessed from the
             context within the notebook.
         io_manager_key (Optional[str]): A string key for the IO manager used to store the output notebook.

dagstermill/context.py

@@ -7,7 +7,7 @@ from dagster import (
     OpDefinition,
     _check as check,
 )
-from dagster._annotations import public
+from dagster._annotations import beta, public
 from dagster._core.definitions.dependency import Node, NodeHandle
 from dagster._core.definitions.repository_definition.repository_definition import (
     RepositoryDefinition,
@@ -18,6 +18,7 @@ from dagster._core.log_manager import DagsterLogManager
 from dagster._core.system_config.objects import ResolvedRunConfig
 
 
+@beta
 class DagstermillExecutionContext(AbstractComputeExecutionContext):
     """Dagstermill-specific execution context.
 

dagstermill/factory.py

@@ -17,6 +17,7 @@ from dagster import (
     _check as check,
     _seven,
 )
+from dagster._annotations import beta
 from dagster._config.pythonic_config import Config, infer_schema_from_config_class
 from dagster._config.pythonic_config.type_check_utils import safe_is_subclass
 from dagster._core.definitions.events import AssetMaterialization, Failure, RetryRequested
@@ -62,6 +63,7 @@ def _find_first_tagged_cell_index(nb, tag):
 # This is based on papermill.parameterize.parameterize_notebook
 # Typically, papermill injects the injected-parameters cell *below* the parameters cell
 # but we want to *replace* the parameters cell, which is what this function does.
+@beta
 def replace_parameters(context, nb, parameters):
     """Assigned parameters into the appropriate place in the input notebook.
 
@@ -108,6 +110,7 @@ def replace_parameters(context, nb, parameters):
     return nb
 
 
+@beta
 def get_papermill_parameters(
     step_context: StepExecutionContext,
     inputs: Mapping[str, object],
@@ -161,6 +164,7 @@ def get_papermill_parameters(
     return parameters
 
 
+@beta
 def execute_notebook(
     step_context: StepExecutionContext,
     name: str,
@@ -342,6 +346,7 @@ def _make_dagstermill_compute_fn(
     return _t_fn
 
 
+@beta
 def define_dagstermill_op(
     name: str,
     notebook_path: str,

dagstermill/io_managers.py

@@ -11,6 +11,7 @@ from dagster import (
     InitResourceContext,
     IOManager,
 )
+from dagster._annotations import beta
 from dagster._core.definitions.metadata import MetadataValue
 from dagster._core.execution.context.input import InputContext
 from dagster._core.execution.context.output import OutputContext
@@ -83,6 +84,7 @@ class LocalOutputNotebookIOManager(OutputNotebookIOManager):
             return file_obj.read()
 
 
+@beta
 class ConfigurableLocalOutputNotebookIOManager(ConfigurableIOManagerFactory):
     """Built-in IO Manager for handling output notebook."""
 
@@ -112,6 +114,7 @@ class ConfigurableLocalOutputNotebookIOManager(ConfigurableIOManagerFactory):
         )
 
 
+@beta
 @dagster_maintained_io_manager
 @io_manager(config_schema=ConfigurableLocalOutputNotebookIOManager.to_config_schema())
 def local_output_notebook_io_manager(init_context) -> LocalOutputNotebookIOManager:

dagstermill/manager.py

@@ -15,6 +15,7 @@ from dagster import (
     TypeCheck,
     _check as check,
 )
+from dagster._annotations import beta
 from dagster._core.definitions.dependency import NodeHandle
 from dagster._core.definitions.events import RetryRequested
 from dagster._core.definitions.graph_definition import GraphDefinition
@@ -64,6 +65,7 @@ class DagstermillResourceEventGenerationManager(EventGenerationManager):
         return [teardown_event for teardown_event in super().generate_teardown_events()]
 
 
+@beta
 class Manager:
     def __init__(self):
         self.job = None

dagstermill/version.py

@@ -1 +1 @@
-__version__ = "0.25.12"
+__version__ = "0.26.0"

{dagstermill-0.25.12.dist-info → dagstermill-0.26.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagstermill
-Version: 0.25.12
+Version: 0.26.0
 Summary: run notebooks using the Dagster tools
 Author: Dagster Labs
 Author-email: hello@dagsterlabs.com
@@ -12,7 +12,7 @@ Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.9,<3.13
 License-File: LICENSE
-Requires-Dist: dagster ==1.9.12
+Requires-Dist: dagster ==1.10.0
 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.26.0.dist-info/RECORD

@@ -0,0 +1,22 @@
+dagstermill/__init__.py,sha256=3YqU9M3bCTeiof-BEHLmZ-A06HEYxdALfSu1pB9QAeg,1262
+dagstermill/__main__.py,sha256=GqkyStw1mh9zw2gMbQBQpIqW0hpr3I8unVWrw_VqdW8,41
+dagstermill/asset_factory.py,sha256=DH5LRZ_2w74G7rgoCNXTUS4HRdIjlXx8MN1ElP1e7Qo,9841
+dagstermill/cli.py,sha256=J8u70wrJ_WvELbEo6Dchiwso6Y3E7r5FuHCcjQb4E4I,4521
+dagstermill/compat.py,sha256=GCfUaGC3eIEhlZP_VFtua8hFKNtwXocOAfYxuKZ5X3I,526
+dagstermill/context.py,sha256=gng48LHtyXbU171WkIfyxS_q8Qgg_ItD6Z2FN625lAo,6171
+dagstermill/engine.py,sha256=exxp3VUOKcTWjEOlc0NQZkxeRPmR3JYTFp7BBH5xM7I,5788
+dagstermill/errors.py,sha256=WOmpAGp-J1XhyGK_LT3ZZKsBwF5dvrWbqSaSldtoh6Y,141
+dagstermill/factory.py,sha256=vwTygRE1FETR_0L7LVRFa0n-wu99c0YTaLm0jMa_FNc,18842
+dagstermill/io_managers.py,sha256=zRIMpZ_j61942pTttrXdwwkS8Qv20x0r3V0JWf1xnSg,4440
+dagstermill/manager.py,sha256=2Hz_I-jsPaZNsxXPRxdpezhhfBD3bnAjrQBz_gdqSlg,16105
+dagstermill/serialize.py,sha256=eXW3c26CiILT_uebyFcAKBnsiNxnjyGT_ch3PfGyjek,188
+dagstermill/translator.py,sha256=h1VPAOWtdjLKzFjRmyN9hO_R6qJuAETNkJydfdgwWGM,2170
+dagstermill/version.py,sha256=S1dNBNSmHJkAHkq1vL1Yg-dFpQk0fri_CstNlL1u5qw,23
+dagstermill/examples/__init__.py,sha256=_2u28VKkHmd16vz0Q2N7Fw5QKsJ65HEnIfJL26gNUIw,75
+dagstermill/examples/repository.py,sha256=AlBcvZHCqZ4wAGp8YjO67A1STx0FPhBG97vGEjZv3Bw,15858
+dagstermill-0.26.0.dist-info/LICENSE,sha256=TMatHW4_G9ldRdodEAp-l2Xa2WvsdeOh60E3v1R2jis,11349
+dagstermill-0.26.0.dist-info/METADATA,sha256=OhMtiEAf7OVOJ3IJMQ9tV5xdlKKrUllzvS0NZCfw66M,936
+dagstermill-0.26.0.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+dagstermill-0.26.0.dist-info/entry_points.txt,sha256=885a7vvhABYWEj7W28elkzSmIcKO3REkdd5h4Z4DEJs,53
+dagstermill-0.26.0.dist-info/top_level.txt,sha256=YDelJKdA5YIIrjsObdd8U4E9YhuXJLRe9NKfUzud9Uc,12
+dagstermill-0.26.0.dist-info/RECORD,,

dagstermill-0.25.12.dist-info/RECORD

@@ -1,22 +0,0 @@
-dagstermill/__init__.py,sha256=3YqU9M3bCTeiof-BEHLmZ-A06HEYxdALfSu1pB9QAeg,1262
-dagstermill/__main__.py,sha256=GqkyStw1mh9zw2gMbQBQpIqW0hpr3I8unVWrw_VqdW8,41
-dagstermill/asset_factory.py,sha256=v-vi5hx4J9u75ihI9HtIAD4D3t08BljuCeqL2LdfSJs,9758
-dagstermill/cli.py,sha256=J8u70wrJ_WvELbEo6Dchiwso6Y3E7r5FuHCcjQb4E4I,4521
-dagstermill/compat.py,sha256=GCfUaGC3eIEhlZP_VFtua8hFKNtwXocOAfYxuKZ5X3I,526
-dagstermill/context.py,sha256=FAR2EjRZaatATl25RIQM7ERHz0SVqQenFTjVa2xsNSk,6159
-dagstermill/engine.py,sha256=exxp3VUOKcTWjEOlc0NQZkxeRPmR3JYTFp7BBH5xM7I,5788
-dagstermill/errors.py,sha256=WOmpAGp-J1XhyGK_LT3ZZKsBwF5dvrWbqSaSldtoh6Y,141
-dagstermill/factory.py,sha256=adgggEb3oAMaEF3DkDQDQ-t460axZVKdQD3MlZPSvxQ,18780
-dagstermill/io_managers.py,sha256=GqDT-T0VJMB301ExYiNT_Dxs3kyMZA8u-oj7YRAlFtg,4390
-dagstermill/manager.py,sha256=pVZl0vLx2-ZHKUlAJkHyveIoVxqSyMWFBqYdfQ3t3oY,16061
-dagstermill/serialize.py,sha256=eXW3c26CiILT_uebyFcAKBnsiNxnjyGT_ch3PfGyjek,188
-dagstermill/translator.py,sha256=h1VPAOWtdjLKzFjRmyN9hO_R6qJuAETNkJydfdgwWGM,2170
-dagstermill/version.py,sha256=ae7JWfpdojMZMzpeaY3EWP2rDq4ceUccvsCEiGI6MEI,24
-dagstermill/examples/__init__.py,sha256=_2u28VKkHmd16vz0Q2N7Fw5QKsJ65HEnIfJL26gNUIw,75
-dagstermill/examples/repository.py,sha256=AlBcvZHCqZ4wAGp8YjO67A1STx0FPhBG97vGEjZv3Bw,15858
-dagstermill-0.25.12.dist-info/LICENSE,sha256=TMatHW4_G9ldRdodEAp-l2Xa2WvsdeOh60E3v1R2jis,11349
-dagstermill-0.25.12.dist-info/METADATA,sha256=-5c7rXIhftMzy7XuL7keZU5NmvbV1MEelEyEkKiLlIw,937
-dagstermill-0.25.12.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-dagstermill-0.25.12.dist-info/entry_points.txt,sha256=885a7vvhABYWEj7W28elkzSmIcKO3REkdd5h4Z4DEJs,53
-dagstermill-0.25.12.dist-info/top_level.txt,sha256=YDelJKdA5YIIrjsObdd8U4E9YhuXJLRe9NKfUzud9Uc,12
-dagstermill-0.25.12.dist-info/RECORD,,