dagstermill 0.25.13__tar.gz → 0.26.0__tar.gz

{dagstermill-0.25.13/dagstermill.egg-info → dagstermill-0.26.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagstermill
-Version: 0.25.13
+Version: 0.26.0
 Summary: run notebooks using the Dagster tools
 Author: Dagster Labs
 Author-email: hello@dagsterlabs.com

{dagstermill-0.25.13 → dagstermill-0.26.0}/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-0.25.13 → dagstermill-0.26.0}/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-0.25.13 → dagstermill-0.26.0}/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-0.25.13 → dagstermill-0.26.0}/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-0.25.13 → dagstermill-0.26.0}/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-0.26.0/dagstermill/version.py

@@ -0,0 +1 @@
+__version__ = "0.26.0"

{dagstermill-0.25.13 → dagstermill-0.26.0/dagstermill.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagstermill
-Version: 0.25.13
+Version: 0.26.0
 Summary: run notebooks using the Dagster tools
 Author: Dagster Labs
 Author-email: hello@dagsterlabs.com

{dagstermill-0.25.13 → dagstermill-0.26.0}/dagstermill.egg-info/requires.txt

@@ -1,4 +1,4 @@
-dagster==1.9.13
+dagster==1.10.0
 ipykernel!=5.4.0,!=5.4.1,>=4.9.0
 ipython_genutils>=0.2.0
 packaging>=20.9

{dagstermill-0.25.13 → dagstermill-0.26.0}/setup.py

@@ -31,7 +31,7 @@ setup(
     ],
     python_requires=">=3.9,<3.13",
     install_requires=[
-        "dagster==1.9.13",
+        "dagster==1.10.0",
         # ipykernel 5.4.0 and 5.4.1 broke papermill
         # see https://github.com/dagster-io/dagster/issues/3401,
         # https://github.com/nteract/papermill/issues/519,

dagstermill-0.25.13/dagstermill/version.py

@@ -1 +0,0 @@
-__version__ = "0.25.13"