dagster-wandb 0.22.14__tar.gz → 0.23.1__tar.gz

{dagster-wandb-0.22.14/dagster_wandb.egg-info → dagster-wandb-0.23.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-wandb
-Version: 0.22.14
+Version: 0.23.1
 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.22.14 → dagster-wandb-0.23.1}/dagster_wandb/io_manager.py

@@ -44,6 +44,8 @@ if sys.version_info >= (3, 8):
 else:
     from typing_extensions import TypedDict
 
+UNIT_TEST_RUN_ID = "0ab2e48b-6d63-4ff5-b160-662cc60145f4"
+
 
 class Config(TypedDict):
     dagster_run_id: str
@@ -161,8 +163,8 @@ class ArtifactsIOManager(IOManager):
 
         with self.wandb_run() as run:
             parameters = {}
-            if context.metadata is not None:
-                parameters = context.metadata.get("wandb_artifact_configuration", {})
+            if context.definition_metadata is not None:
+                parameters = context.definition_metadata.get("wandb_artifact_configuration", {})
 
             raise_on_unknown_write_configuration_keys(parameters)
 
@@ -364,8 +366,8 @@ class ArtifactsIOManager(IOManager):
     def _download_artifact(self, context: InputContext):
         with self.wandb_run() as run:
             parameters = {}
-            if context.metadata is not None:
-                parameters = context.metadata.get("wandb_artifact_configuration", {})
+            if context.definition_metadata is not None:
+                parameters = context.definition_metadata.get("wandb_artifact_configuration", {})
 
             raise_on_unknown_read_configuration_keys(parameters)
 
@@ -711,7 +713,7 @@ def wandb_artifacts_io_manager(context: InitResourceContext):
         cache_duration_in_minutes = context.resource_config.get("cache_duration_in_minutes")
 
     if "PYTEST_CURRENT_TEST" in os.environ:
-        dagster_run_id = "unit-testing"
+        dagster_run_id = UNIT_TEST_RUN_ID
     else:
         dagster_run_id = context.run_id
 

dagster-wandb-0.23.1/dagster_wandb/version.py

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

{dagster-wandb-0.22.14 → dagster-wandb-0.23.1/dagster_wandb.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dagster-wandb
-Version: 0.22.14
+Version: 0.23.1
 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.22.14 → dagster-wandb-0.23.1}/dagster_wandb.egg-info/requires.txt

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

{dagster-wandb-0.22.14 → dagster-wandb-0.23.1}/setup.py

@@ -34,7 +34,7 @@ setup(
     packages=find_packages(exclude=["dagster_wandb_tests*"]),
     python_requires=">=3.8,<3.13",
     install_requires=[
-        "dagster==1.6.14",
+        "dagster==1.7.1",
         "wandb>=0.15.11,<1.0",
     ],
     extras_require={"dev": ["cloudpickle", "joblib", "callee", "dill"]},

dagster-wandb-0.22.14/dagster_wandb/version.py

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