dagstermill 0.25.8__py3-none-any.whl → 0.25.9__py3-none-any.whl

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/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/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/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/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/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/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/version.py

@@ -1 +1 @@
-__version__ = "0.25.8"
+__version__ = "0.25.9"

{dagstermill-0.25.8.dist-info → dagstermill-0.25.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagstermill
-Version: 0.25.8
+Version: 0.25.9
 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.8
+Requires-Dist: dagster ==1.9.9
 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.25.9.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=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=NsKiCCQq5j7wW1paL-Bw27h63w_P0r0bIHvsX9TsjGY,23
+dagstermill/examples/__init__.py,sha256=_2u28VKkHmd16vz0Q2N7Fw5QKsJ65HEnIfJL26gNUIw,75
+dagstermill/examples/repository.py,sha256=AlBcvZHCqZ4wAGp8YjO67A1STx0FPhBG97vGEjZv3Bw,15858
+dagstermill-0.25.9.dist-info/LICENSE,sha256=TMatHW4_G9ldRdodEAp-l2Xa2WvsdeOh60E3v1R2jis,11349
+dagstermill-0.25.9.dist-info/METADATA,sha256=2Qr6hKd3LcgW_y580SI3pujLmqwAB1SseNvXCFMPqF8,935
+dagstermill-0.25.9.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
+dagstermill-0.25.9.dist-info/entry_points.txt,sha256=885a7vvhABYWEj7W28elkzSmIcKO3REkdd5h4Z4DEJs,53
+dagstermill-0.25.9.dist-info/top_level.txt,sha256=YDelJKdA5YIIrjsObdd8U4E9YhuXJLRe9NKfUzud9Uc,12
+dagstermill-0.25.9.dist-info/RECORD,,

dagstermill-0.25.8.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=AoUd0lvZT4xAiuC0Sfl7ChtRCYI0yjbipLO-YLA3RDI,9742
-dagstermill/cli.py,sha256=lffeaQgenSR7s8MhNUBNA_BeVehlMUrdg7gnQl1WTdY,4494
-dagstermill/compat.py,sha256=GCfUaGC3eIEhlZP_VFtua8hFKNtwXocOAfYxuKZ5X3I,526
-dagstermill/context.py,sha256=4NwSe11zYhhi35L-uevIZfZlxxoxoVS1Eh4kHS3Yeqs,6117
-dagstermill/engine.py,sha256=RI1jAWaPR2Hi-r09qRRPHUHxbnUMMys3K63upI7BS6o,5854
-dagstermill/errors.py,sha256=WOmpAGp-J1XhyGK_LT3ZZKsBwF5dvrWbqSaSldtoh6Y,141
-dagstermill/factory.py,sha256=gs-Hei2gijrbVc3G2smb_akUVwDTmyItYybZx4Zy8io,18764
-dagstermill/io_managers.py,sha256=D9UA4KufzrOeCan3TJyPogD0uIJBjK80ExikdMHqweI,4403
-dagstermill/manager.py,sha256=gJhoqonXcXjN7jWB1P0THMdfRuKoKV7uHKD6ITsTr6o,16130
-dagstermill/serialize.py,sha256=eXW3c26CiILT_uebyFcAKBnsiNxnjyGT_ch3PfGyjek,188
-dagstermill/translator.py,sha256=h1VPAOWtdjLKzFjRmyN9hO_R6qJuAETNkJydfdgwWGM,2170
-dagstermill/version.py,sha256=bkf--LdvzFbtaf_d7GSQn5A2t-yDZ-SX33wKaaeqHsY,23
-dagstermill/examples/__init__.py,sha256=_2u28VKkHmd16vz0Q2N7Fw5QKsJ65HEnIfJL26gNUIw,75
-dagstermill/examples/repository.py,sha256=AlBcvZHCqZ4wAGp8YjO67A1STx0FPhBG97vGEjZv3Bw,15858
-dagstermill-0.25.8.dist-info/LICENSE,sha256=TMatHW4_G9ldRdodEAp-l2Xa2WvsdeOh60E3v1R2jis,11349
-dagstermill-0.25.8.dist-info/METADATA,sha256=54sUgcV2Fy4LKzusr_bAhUeb26-y7rroh9j4WfUdEWU,935
-dagstermill-0.25.8.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
-dagstermill-0.25.8.dist-info/entry_points.txt,sha256=885a7vvhABYWEj7W28elkzSmIcKO3REkdd5h4Z4DEJs,53
-dagstermill-0.25.8.dist-info/top_level.txt,sha256=YDelJKdA5YIIrjsObdd8U4E9YhuXJLRe9NKfUzud9Uc,12
-dagstermill-0.25.8.dist-info/RECORD,,