dagster-wandb 0.25.1__tar.gz → 0.25.10__tar.gz

{dagster-wandb-0.25.1/dagster_wandb.egg-info → dagster-wandb-0.25.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-wandb
-Version: 0.25.1
+Version: 0.25.10
 Summary: Package for wandb Dagster components.
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-wandb
 Author: Dagster Labs

{dagster-wandb-0.25.1 → dagster-wandb-0.25.10}/dagster_wandb/io_manager.py

@@ -3,11 +3,10 @@ import os
 import pickle
 import platform
 import shutil
-import sys
 import time
 import uuid
 from contextlib import contextmanager
-from typing import List, Optional
+from typing import Optional, TypedDict
 
 from dagster import (
     Field,
@@ -39,11 +38,6 @@ from dagster_wandb.utils.pickling import (
 )
 from dagster_wandb.version import __version__
 
-if sys.version_info >= (3, 8):
-    from typing import TypedDict
-else:
-    from typing_extensions import TypedDict
-
 UNIT_TEST_RUN_ID = "0ab2e48b-6d63-4ff5-b160-662cc60145f4"
 
 
@@ -54,7 +48,7 @@ class Config(TypedDict):
     wandb_project: str
     wandb_run_name: Optional[str]
     wandb_run_id: Optional[str]
-    wandb_run_tags: Optional[List[str]]
+    wandb_run_tags: Optional[list[str]]
     base_dir: str
     cache_duration_in_minutes: Optional[int]
 
@@ -295,7 +289,7 @@ class ArtifactsIOManager(IOManager):
                         context.log.warning(
                             "You've included a 'serialization_module' in the"
                             " 'wandb_artifact_configuration' settings. However, this doesn't have"
-                            " any impact when the output is already an W&B object like e.g Table or"
+                            " any impact when the output is already a W&B object like e.g Table or"
                             " Image."
                         )
                     # Adds the WBValue object using the class name as the name for the file

{dagster-wandb-0.25.1 → dagster-wandb-0.25.10}/dagster_wandb/resources.py

@@ -1,4 +1,4 @@
-from typing import Any, Dict
+from typing import Any
 
 import wandb
 from dagster import Field, InitResourceContext, String, StringSource, resource
@@ -27,7 +27,7 @@ WANDB_CLOUD_HOST: str = "https://api.wandb.ai"
     },
     description="Resource for interacting with Weights & Biases",
 )
-def wandb_resource(context: InitResourceContext) -> Dict[str, Any]:
+def wandb_resource(context: InitResourceContext) -> dict[str, Any]:
     """Dagster resource used to communicate with the W&B API. It's useful when you want to use the
     wandb client within your ops and assets. It's a required resources if you are using the W&B IO
     Manager.

{dagster-wandb-0.25.1 → dagster-wandb-0.25.10}/dagster_wandb/types.py

@@ -1,18 +1,11 @@
-import sys
-
-if sys.version_info >= (3, 8):
-    from typing import TypedDict
-else:
-    from typing_extensions import TypedDict
-
-from typing import Any, Dict, List
+from typing import Any, TypedDict
 
 
 class SerializationModule(TypedDict, total=False):
     """W&B Artifacts IO Manager configuration of the serialization module. Useful for type checking."""
 
     name: str
-    parameters: Dict[str, Any]
+    parameters: dict[str, Any]
 
 
 class WandbArtifactConfiguration(TypedDict, total=False):
@@ -21,9 +14,9 @@ class WandbArtifactConfiguration(TypedDict, total=False):
     name: str
     type: str
     description: str
-    aliases: List[str]
-    add_dirs: List[Dict[str, Any]]
-    add_files: List[Dict[str, Any]]
-    add_references: List[Dict[str, Any]]
+    aliases: list[str]
+    add_dirs: list[dict[str, Any]]
+    add_files: list[dict[str, Any]]
+    add_references: list[dict[str, Any]]
     serialization_module: SerializationModule
-    partitions: Dict[str, Dict[str, Any]]
+    partitions: dict[str, dict[str, Any]]

{dagster-wandb-0.25.1 → dagster-wandb-0.25.10}/dagster_wandb/utils/pickling.py

@@ -55,7 +55,7 @@ def pickle_artifact_content(
             **artifact.metadata,
             **{
                 "source_serialization_module": "dill",
-                "source_dill_version_used": dill.__version__,
+                "source_dill_version_used": dill.__version__,  # pyright: ignore[reportOptionalMemberAccess]
                 "source_pickle_protocol_used": serialization_module_parameters_with_protocol[
                     "protocol"
                 ],
@@ -63,7 +63,7 @@ def pickle_artifact_content(
         }
         with artifact.new_file(DILL_FILENAME, "wb") as file:
             try:
-                dill.dump(
+                dill.dump(  # pyright: ignore[reportOptionalMemberAccess]
                     obj,
                     file,
                     **serialization_module_parameters_with_protocol,
@@ -88,7 +88,7 @@ def pickle_artifact_content(
             **artifact.metadata,
             **{
                 "source_serialization_module": "cloudpickle",
-                "source_cloudpickle_version_used": cloudpickle.__version__,
+                "source_cloudpickle_version_used": cloudpickle.__version__,  # pyright: ignore[reportOptionalMemberAccess]
                 "source_pickle_protocol_used": serialization_module_parameters_with_protocol[
                     "protocol"
                 ],
@@ -96,7 +96,7 @@ def pickle_artifact_content(
         }
         with artifact.new_file(CLOUDPICKLE_FILENAME, "wb") as file:
             try:
-                cloudpickle.dump(
+                cloudpickle.dump(  # pyright: ignore[reportOptionalMemberAccess]
                     obj,
                     file,
                     **serialization_module_parameters_with_protocol,
@@ -120,7 +120,7 @@ def pickle_artifact_content(
             **artifact.metadata,
             **{
                 "source_serialization_module": "joblib",
-                "source_joblib_version_used": joblib.__version__,
+                "source_joblib_version_used": joblib.__version__,  # pyright: ignore[reportOptionalMemberAccess]
                 "source_pickle_protocol_used": serialization_module_parameters_with_protocol[
                     "protocol"
                 ],
@@ -128,7 +128,7 @@ def pickle_artifact_content(
         }
         with artifact.new_file(JOBLIB_FILENAME, "wb") as file:
             try:
-                joblib.dump(
+                joblib.dump(  # pyright: ignore[reportOptionalMemberAccess]
                     obj,
                     file,
                     **serialization_module_parameters_with_protocol,
@@ -182,7 +182,7 @@ def unpickle_artifact_content(artifact_dir):
                 " was not found. Please, make sure it's installed."
             )
         with open(f"{artifact_dir}/{DILL_FILENAME}", "rb") as file:
-            input_value = dill.load(file)
+            input_value = dill.load(file)  # pyright: ignore[reportOptionalMemberAccess]
             return input_value
     elif os.path.exists(f"{artifact_dir}/{CLOUDPICKLE_FILENAME}"):
         if not has_cloudpickle:
@@ -191,7 +191,7 @@ def unpickle_artifact_content(artifact_dir):
                 " module was not found. Please, make sure it's installed."
             )
         with open(f"{artifact_dir}/{CLOUDPICKLE_FILENAME}", "rb") as file:
-            input_value = cloudpickle.load(file)
+            input_value = cloudpickle.load(file)  # pyright: ignore[reportOptionalMemberAccess]
             return input_value
     elif os.path.exists(f"{artifact_dir}/{JOBLIB_FILENAME}"):
         if not has_joblib:
@@ -200,7 +200,7 @@ def unpickle_artifact_content(artifact_dir):
                 " was not found. Please, make sure it's installed."
             )
         with open(f"{artifact_dir}/{JOBLIB_FILENAME}", "rb") as file:
-            input_value = joblib.load(file)
+            input_value = joblib.load(file)  # pyright: ignore[reportOptionalMemberAccess]
             return input_value
     elif os.path.exists(f"{artifact_dir}/{PICKLE_FILENAME}"):
         with open(f"{artifact_dir}/{PICKLE_FILENAME}", "rb") as file:

dagster-wandb-0.25.10/dagster_wandb/version.py

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

{dagster-wandb-0.25.1 → dagster-wandb-0.25.10/dagster_wandb.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-wandb
-Version: 0.25.1
+Version: 0.25.10
 Summary: Package for wandb Dagster components.
 Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-wandb
 Author: Dagster Labs

{dagster-wandb-0.25.1 → dagster-wandb-0.25.10}/dagster_wandb.egg-info/requires.txt

@@ -1,4 +1,4 @@
-dagster==1.9.1
+dagster==1.9.10
 wandb<1.0,>=0.15.11
 
 [dev]

{dagster-wandb-0.25.1 → dagster-wandb-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 / "dagster_wandb/version.py", encoding="utf8") as fp:
         exec(fp.read(), version)
 
@@ -34,7 +33,7 @@ setup(
     include_package_data=True,
     python_requires=">=3.9,<3.13",
     install_requires=[
-        "dagster==1.9.1",
+        "dagster==1.9.10",
         "wandb>=0.15.11,<1.0",
     ],
     extras_require={"dev": ["cloudpickle", "joblib", "callee", "dill"]},

dagster-wandb-0.25.1/dagster_wandb/version.py

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

{dagster-wandb-0.25.1 → dagster-wandb-0.25.10}/dagster_wandb/__init__.py

@@ -9,11 +9,11 @@ from dagster_wandb.version import __version__
 DagsterLibraryRegistry.register("dagster-wandb", __version__)
 
 __all__ = [
-    "WandbArtifactsIOManagerError",
     "SerializationModule",
-    "wandb_resource",
-    "wandb_artifacts_io_manager",
     "WandbArtifactConfiguration",
+    "WandbArtifactsIOManagerError",
     "run_launch_agent",
     "run_launch_job",
+    "wandb_artifacts_io_manager",
+    "wandb_resource",
 ]