dagstermill 0.25.8__tar.gz → 0.25.10__tar.gz

{dagstermill-0.25.8/dagstermill.egg-info → dagstermill-0.25.10}/PKG-INFO

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

{dagstermill-0.25.8 → dagstermill-0.25.10}/dagstermill/asset_factory.py

@@ -1,6 +1,7 @@
 import pickle
 import tempfile
-from typing import Any, Callable, Iterable, Mapping, Optional, Set, Type, Union, cast
+from collections.abc import Iterable, Mapping
+from typing import Any, Callable, Optional, Union, cast
 
 import dagster._check as check
 from dagster import (
@@ -76,7 +77,7 @@ def define_dagstermill_asset(
     deps: Optional[Iterable[Union[CoercibleToAssetKey, AssetsDefinition, SourceAsset]]] = None,
     metadata: Optional[Mapping[str, Any]] = None,
     config_schema: Optional[Union[Any, Mapping[str, Any]]] = None,
-    required_resource_keys: Optional[Set[str]] = None,
+    required_resource_keys: Optional[set[str]] = None,
     resource_defs: Optional[Mapping[str, ResourceDefinition]] = None,
     description: Optional[str] = None,
     partitions_def: Optional[PartitionsDefinition] = None,
@@ -85,7 +86,7 @@ def define_dagstermill_asset(
     io_manager_key: Optional[str] = None,
     retry_policy: Optional[RetryPolicy] = None,
     save_notebook_on_failure: bool = False,
-    non_argument_deps: Optional[Union[Set[AssetKey], Set[str]]] = None,
+    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.
@@ -194,7 +195,7 @@ def define_dagstermill_asset(
     }
 
     if safe_is_subclass(config_schema, Config):
-        config_schema = infer_schema_from_config_class(cast(Type[Config], config_schema))
+        config_schema = infer_schema_from_config_class(cast(type[Config], config_schema))
 
     return asset(
         name=name,

{dagstermill-0.25.8 → dagstermill-0.25.10}/dagstermill/cli.py

@@ -1,7 +1,8 @@
 import copy
 import os
 import subprocess
-from typing import Mapping, Optional
+from collections.abc import Mapping
+from typing import Optional
 
 import click
 import dagster._check as check

{dagstermill-0.25.8 → dagstermill-0.25.10}/dagstermill/context.py

@@ -1,4 +1,5 @@
-from typing import AbstractSet, Any, Mapping, Optional, cast
+from collections.abc import Mapping
+from typing import AbstractSet, Any, Optional, cast  # noqa: UP035
 
 from dagster import (
     DagsterRun,

{dagstermill-0.25.8 → dagstermill-0.25.10}/dagstermill/engine.py

@@ -86,9 +86,7 @@ else:
         # the kernel down. Note that atexit doesn't seem to work at all in ipython, and hooking into
         # the ipython post_execute event doesn't work in papermill.
         def papermill_process(self, nb_man, resources):
-            _, resources = super(DagstermillExecutePreprocessor, self).papermill_process(
-                nb_man, resources
-            )
+            _, resources = super().papermill_process(nb_man, resources)
 
             new_cell = nbformat.v4.new_code_cell(
                 source="import dagstermill as __dm_dagstermill\n__dm_dagstermill._teardown()\n"

{dagstermill-0.25.8 → dagstermill-0.25.10}/dagstermill/factory.py

@@ -4,7 +4,8 @@ import pickle
 import sys
 import tempfile
 import uuid
-from typing import Any, Callable, Iterable, Mapping, Optional, Sequence, Set, Type, Union, cast
+from collections.abc import Iterable, Mapping, Sequence
+from typing import Any, Callable, Optional, Union, cast
 
 import nbformat
 import papermill
@@ -347,7 +348,7 @@ def define_dagstermill_op(
     ins: Optional[Mapping[str, In]] = None,
     outs: Optional[Mapping[str, Out]] = None,
     config_schema: Optional[Union[Any, Mapping[str, Any]]] = None,
-    required_resource_keys: Optional[Set[str]] = None,
+    required_resource_keys: Optional[set[str]] = None,
     output_notebook_name: Optional[str] = None,
     asset_key_prefix: Optional[Union[Sequence[str], str]] = None,
     description: Optional[str] = None,
@@ -431,7 +432,7 @@ def define_dagstermill_op(
     }
 
     if safe_is_subclass(config_schema, Config):
-        config_schema = infer_schema_from_config_class(cast(Type[Config], config_schema))
+        config_schema = infer_schema_from_config_class(cast(type[Config], config_schema))
 
     return OpDefinition(
         name=name,

{dagstermill-0.25.8 → dagstermill-0.25.10}/dagstermill/io_managers.py

@@ -1,6 +1,7 @@
 import os
+from collections.abc import Sequence
 from pathlib import Path
-from typing import Any, List, Optional, Sequence
+from typing import Any, Optional
 
 import dagster._check as check
 from dagster import (
@@ -33,7 +34,7 @@ class OutputNotebookIOManager(IOManager):
 
 class LocalOutputNotebookIOManager(OutputNotebookIOManager):
     def __init__(self, base_dir: str, asset_key_prefix: Optional[Sequence[str]] = None):
-        super(LocalOutputNotebookIOManager, self).__init__(asset_key_prefix=asset_key_prefix)
+        super().__init__(asset_key_prefix=asset_key_prefix)
         self.base_dir = base_dir
         self.write_mode = "wb"
         self.read_mode = "rb"
@@ -92,7 +93,7 @@ class ConfigurableLocalOutputNotebookIOManager(ConfigurableIOManagerFactory):
             " directory if not provided."
         ),
     )
-    asset_key_prefix: List[str] = Field(
+    asset_key_prefix: list[str] = Field(
         default=[],
         description=(
             "Asset key prefix to apply to assets materialized for output notebooks. Defaults to no"

{dagstermill-0.25.8 → dagstermill-0.25.10}/dagstermill/manager.py

@@ -1,7 +1,8 @@
 import os
 import pickle
 import uuid
-from typing import TYPE_CHECKING, AbstractSet, Any, Mapping, Optional, cast
+from collections.abc import Mapping
+from typing import TYPE_CHECKING, AbstractSet, Any, Optional, cast  # noqa: UP035
 
 from dagster import (
     AssetMaterialization,
@@ -60,12 +61,7 @@ class DagstermillResourceEventGenerationManager(EventGenerationManager):
         return iter(())
 
     def teardown(self):
-        return [
-            teardown_event
-            for teardown_event in super(
-                DagstermillResourceEventGenerationManager, self
-            ).generate_teardown_events()
-        ]
+        return [teardown_event for teardown_event in super().generate_teardown_events()]
 
 
 class Manager:

dagstermill-0.25.10/dagstermill/version.py

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

{dagstermill-0.25.8 → dagstermill-0.25.10/dagstermill.egg-info}/PKG-INFO

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

{dagstermill-0.25.8 → dagstermill-0.25.10}/dagstermill.egg-info/requires.txt

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

{dagstermill-0.25.8 → dagstermill-0.25.10}/setup.py

@@ -1,11 +1,10 @@
 from pathlib import Path
-from typing import Dict
 
 from setuptools import find_packages, setup
 
 
 def get_version() -> str:
-    version: Dict[str, str] = {}
+    version: dict[str, str] = {}
     with open(Path(__file__).parent / "dagstermill/version.py", encoding="utf8") as fp:
         exec(fp.read(), version)
 
@@ -32,7 +31,7 @@ setup(
     ],
     python_requires=">=3.9,<3.13",
     install_requires=[
-        "dagster==1.9.8",
+        "dagster==1.9.10",
         # 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.8/dagstermill/version.py

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