dagster-wandb 0.22.14__py3-none-any.whl → 0.23.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of dagster-wandb might be problematic. Click here for more details.

Files changed (7) hide show
  1. dagster_wandb/io_manager.py +7 -5
  2. dagster_wandb/version.py +1 -1
  3. {dagster_wandb-0.22.14.dist-info → dagster_wandb-0.23.1.dist-info}/METADATA +2 -2
  4. {dagster_wandb-0.22.14.dist-info → dagster_wandb-0.23.1.dist-info}/RECORD +7 -7
  5. {dagster_wandb-0.22.14.dist-info → dagster_wandb-0.23.1.dist-info}/LICENSE +0 -0
  6. {dagster_wandb-0.22.14.dist-info → dagster_wandb-0.23.1.dist-info}/WHEEL +0 -0
  7. {dagster_wandb-0.22.14.dist-info → dagster_wandb-0.23.1.dist-info}/top_level.txt +0 -0
dagster_wandb/io_manager.py CHANGED
@@ -44,6 +44,8 @@ if sys.version_info >= (3, 8):
44
44
  else:
45
45
  from typing_extensions import TypedDict
46
46
 
47
+ UNIT_TEST_RUN_ID = "0ab2e48b-6d63-4ff5-b160-662cc60145f4"
48
+
47
49
 
48
50
  class Config(TypedDict):
49
51
  dagster_run_id: str
@@ -161,8 +163,8 @@ class ArtifactsIOManager(IOManager):
161
163
 
162
164
  with self.wandb_run() as run:
163
165
  parameters = {}
164
- if context.metadata is not None:
165
- parameters = context.metadata.get("wandb_artifact_configuration", {})
166
+ if context.definition_metadata is not None:
167
+ parameters = context.definition_metadata.get("wandb_artifact_configuration", {})
166
168
 
167
169
  raise_on_unknown_write_configuration_keys(parameters)
168
170
 
@@ -364,8 +366,8 @@ class ArtifactsIOManager(IOManager):
364
366
  def _download_artifact(self, context: InputContext):
365
367
  with self.wandb_run() as run:
366
368
  parameters = {}
367
- if context.metadata is not None:
368
- parameters = context.metadata.get("wandb_artifact_configuration", {})
369
+ if context.definition_metadata is not None:
370
+ parameters = context.definition_metadata.get("wandb_artifact_configuration", {})
369
371
 
370
372
  raise_on_unknown_read_configuration_keys(parameters)
371
373
 
@@ -711,7 +713,7 @@ def wandb_artifacts_io_manager(context: InitResourceContext):
711
713
  cache_duration_in_minutes = context.resource_config.get("cache_duration_in_minutes")
712
714
 
713
715
  if "PYTEST_CURRENT_TEST" in os.environ:
714
- dagster_run_id = "unit-testing"
716
+ dagster_run_id = UNIT_TEST_RUN_ID
715
717
  else:
716
718
  dagster_run_id = context.run_id
717
719
 
dagster_wandb/version.py CHANGED
@@ -1 +1 @@
1
- __version__ = "0.22.14"
1
+ __version__ = "0.23.1"
{dagster_wandb-0.22.14.dist-info → dagster_wandb-0.23.1.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: dagster-wandb
3
- Version: 0.22.14
3
+ Version: 0.23.1
4
4
  Summary: Package for wandb Dagster components.
5
5
  Home-page: https://github.com/dagster-io/dagster/tree/master/python_modules/libraries/dagster-wandb
6
6
  Author: Dagster Labs
@@ -14,7 +14,7 @@ Classifier: License :: OSI Approved :: Apache Software License
14
14
  Classifier: Operating System :: OS Independent
15
15
  Requires-Python: >=3.8,<3.13
16
16
  License-File: LICENSE
17
- Requires-Dist: dagster ==1.6.14
17
+ Requires-Dist: dagster ==1.7.1
18
18
  Requires-Dist: wandb <1.0,>=0.15.11
19
19
  Provides-Extra: dev
20
20
  Requires-Dist: cloudpickle ; extra == 'dev'
{dagster_wandb-0.22.14.dist-info → dagster_wandb-0.23.1.dist-info}/RECORD RENAMED
@@ -1,16 +1,16 @@
1
1
  dagster_wandb/__init__.py,sha256=-rUAd_xLZvcoA6d7sERb_xxkIThb0QiFan61C6EB8A0,613
2
- dagster_wandb/io_manager.py,sha256=PWFqUr1bnECkmIXmzUdIGKxjs1HBZ68r6pq6NHQZ3ps,33981
2
+ dagster_wandb/io_manager.py,sha256=dNIndjRZUhutD29pnJPTZg9WuB7xVqI9uuHdFmJ487w,34086
3
3
  dagster_wandb/resources.py,sha256=uWqNCxoiQaM6NeavJxKul8Qu_pkSmKb-bqbEDOmghd8,2272
4
4
  dagster_wandb/types.py,sha256=nB61MN1Lsny91Ahli16WPC628jTlC0V95tI6DsXW3y0,774
5
- dagster_wandb/version.py,sha256=jyda_cqSZbwexWt3ATGl7DuFA5PyH-kDQX7yuu4V4Rk,24
5
+ dagster_wandb/version.py,sha256=EmtVqV41YmZp6Mj5pkm4iGXDMznVEE4GQEUYGGidq30,23
6
6
  dagster_wandb/launch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
7
  dagster_wandb/launch/configs.py,sha256=bA7x8LfE2yc3UEK3eUM_NlCgix7cUY4ziISiRaVKjww,4425
8
8
  dagster_wandb/launch/ops.py,sha256=Ib4d7aWmYbyIYwVcjFTehASydcsM7YlbjCwRt0Xv1-M,5153
9
9
  dagster_wandb/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
10
10
  dagster_wandb/utils/errors.py,sha256=Ow4xy8n1pZp6_RC_WzxQVrD9FCEtc_mnF6k7q8jMT2Y,2744
11
11
  dagster_wandb/utils/pickling.py,sha256=-c3TXZqFnWEZ4zbxY3Odba5HIAjlUS3YpSEeZ1Icpuw,7890
12
- dagster_wandb-0.22.14.dist-info/LICENSE,sha256=tAkwu8-AdEyGxGoSvJ2gVmQdcicWw3j1ZZueVV74M-E,11357
13
- dagster_wandb-0.22.14.dist-info/METADATA,sha256=MklLJgEgH6oxWKcTaWhjT1SE3VRlSykVO8YYoVkBRa4,894
14
- dagster_wandb-0.22.14.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
15
- dagster_wandb-0.22.14.dist-info/top_level.txt,sha256=eBR6kNspYqHANL4Hy1aXRhq-pYZmPhFg_xKui2AS8cE,14
16
- dagster_wandb-0.22.14.dist-info/RECORD,,
12
+ dagster_wandb-0.23.1.dist-info/LICENSE,sha256=tAkwu8-AdEyGxGoSvJ2gVmQdcicWw3j1ZZueVV74M-E,11357
13
+ dagster_wandb-0.23.1.dist-info/METADATA,sha256=XP9DuxSv7yanArm6iknQw43mwKnG4esXs3T2j3ulHho,892
14
+ dagster_wandb-0.23.1.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
15
+ dagster_wandb-0.23.1.dist-info/top_level.txt,sha256=eBR6kNspYqHANL4Hy1aXRhq-pYZmPhFg_xKui2AS8cE,14
16
+ dagster_wandb-0.23.1.dist-info/RECORD,,