dagster-wandb 0.25.8__py3-none-any.whl → 0.25.9__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.

@@ -3,11 +3,10 @@ import os
3
3
  import pickle
4
4
  import platform
5
5
  import shutil
6
- import sys
7
6
  import time
8
7
  import uuid
9
8
  from contextlib import contextmanager
10
- from typing import List, Optional
9
+ from typing import Optional, TypedDict
11
10
 
12
11
  from dagster import (
13
12
  Field,
@@ -39,11 +38,6 @@ from dagster_wandb.utils.pickling import (
39
38
  )
40
39
  from dagster_wandb.version import __version__
41
40
 
42
- if sys.version_info >= (3, 8):
43
- from typing import TypedDict
44
- else:
45
- from typing_extensions import TypedDict
46
-
47
41
  UNIT_TEST_RUN_ID = "0ab2e48b-6d63-4ff5-b160-662cc60145f4"
48
42
 
49
43
 
@@ -54,7 +48,7 @@ class Config(TypedDict):
54
48
  wandb_project: str
55
49
  wandb_run_name: Optional[str]
56
50
  wandb_run_id: Optional[str]
57
- wandb_run_tags: Optional[List[str]]
51
+ wandb_run_tags: Optional[list[str]]
58
52
  base_dir: str
59
53
  cache_duration_in_minutes: Optional[int]
60
54
 
@@ -1,4 +1,4 @@
1
- from typing import Any, Dict
1
+ from typing import Any
2
2
 
3
3
  import wandb
4
4
  from dagster import Field, InitResourceContext, String, StringSource, resource
@@ -27,7 +27,7 @@ WANDB_CLOUD_HOST: str = "https://api.wandb.ai"
27
27
  },
28
28
  description="Resource for interacting with Weights & Biases",
29
29
  )
30
- def wandb_resource(context: InitResourceContext) -> Dict[str, Any]:
30
+ def wandb_resource(context: InitResourceContext) -> dict[str, Any]:
31
31
  """Dagster resource used to communicate with the W&B API. It's useful when you want to use the
32
32
  wandb client within your ops and assets. It's a required resources if you are using the W&B IO
33
33
  Manager.
dagster_wandb/types.py CHANGED
@@ -1,18 +1,11 @@
1
- import sys
2
-
3
- if sys.version_info >= (3, 8):
4
- from typing import TypedDict
5
- else:
6
- from typing_extensions import TypedDict
7
-
8
- from typing import Any, Dict, List
1
+ from typing import Any, TypedDict
9
2
 
10
3
 
11
4
  class SerializationModule(TypedDict, total=False):
12
5
  """W&B Artifacts IO Manager configuration of the serialization module. Useful for type checking."""
13
6
 
14
7
  name: str
15
- parameters: Dict[str, Any]
8
+ parameters: dict[str, Any]
16
9
 
17
10
 
18
11
  class WandbArtifactConfiguration(TypedDict, total=False):
@@ -21,9 +14,9 @@ class WandbArtifactConfiguration(TypedDict, total=False):
21
14
  name: str
22
15
  type: str
23
16
  description: str
24
- aliases: List[str]
25
- add_dirs: List[Dict[str, Any]]
26
- add_files: List[Dict[str, Any]]
27
- add_references: List[Dict[str, Any]]
17
+ aliases: list[str]
18
+ add_dirs: list[dict[str, Any]]
19
+ add_files: list[dict[str, Any]]
20
+ add_references: list[dict[str, Any]]
28
21
  serialization_module: SerializationModule
29
- partitions: Dict[str, Dict[str, Any]]
22
+ partitions: dict[str, dict[str, Any]]
dagster_wandb/version.py CHANGED
@@ -1 +1 @@
1
- __version__ = "0.25.8"
1
+ __version__ = "0.25.9"
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: dagster-wandb
3
- Version: 0.25.8
3
+ Version: 0.25.9
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
@@ -13,7 +13,7 @@ Classifier: License :: OSI Approved :: Apache Software License
13
13
  Classifier: Operating System :: OS Independent
14
14
  Requires-Python: >=3.9,<3.13
15
15
  License-File: LICENSE
16
- Requires-Dist: dagster ==1.9.8
16
+ Requires-Dist: dagster ==1.9.9
17
17
  Requires-Dist: wandb <1.0,>=0.15.11
18
18
  Provides-Extra: dev
19
19
  Requires-Dist: cloudpickle ; extra == 'dev'
@@ -1,17 +1,17 @@
1
1
  dagster_wandb/__init__.py,sha256=PMLVZ9Gniz3S6RPNLbNg0ljZSHvAH9gjQPF5LbiVh8Y,678
2
- dagster_wandb/io_manager.py,sha256=FbzHBlhTN3r0gpcN-hDuZdMWvfSWYF7Y6nvZOyoPMu0,34107
2
+ dagster_wandb/io_manager.py,sha256=F7N6qieSntyP3EgjwL60wvUhr4Pjycs1xSIhBM5fD8w,33986
3
3
  dagster_wandb/py.typed,sha256=mDShSrm8qg9qjacQc2F-rI8ATllqP6EdgHuEYxuCXZ0,7
4
- dagster_wandb/resources.py,sha256=uWqNCxoiQaM6NeavJxKul8Qu_pkSmKb-bqbEDOmghd8,2272
5
- dagster_wandb/types.py,sha256=nB61MN1Lsny91Ahli16WPC628jTlC0V95tI6DsXW3y0,774
6
- dagster_wandb/version.py,sha256=bkf--LdvzFbtaf_d7GSQn5A2t-yDZ-SX33wKaaeqHsY,23
4
+ dagster_wandb/resources.py,sha256=Fc98agHa6ZUyNunxcwaw1vTu-teK0_HCIQ2Rd8kClcI,2266
5
+ dagster_wandb/types.py,sha256=Stbxy_49lBgk8jmS6ujk0K8AFgnUYZ_S_w5KQLbUAmA,646
6
+ dagster_wandb/version.py,sha256=NsKiCCQq5j7wW1paL-Bw27h63w_P0r0bIHvsX9TsjGY,23
7
7
  dagster_wandb/launch/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
8
8
  dagster_wandb/launch/configs.py,sha256=czir6F-iaGf7j9ur-I9Mi-MwpWkf--O-xx4SJK3Iehk,4421
9
9
  dagster_wandb/launch/ops.py,sha256=mgJaPrC91jaUVRiXmMtB1Xzk5MbWxEV1gZdUH1dMt-g,5173
10
10
  dagster_wandb/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
11
11
  dagster_wandb/utils/errors.py,sha256=Ow4xy8n1pZp6_RC_WzxQVrD9FCEtc_mnF6k7q8jMT2Y,2744
12
12
  dagster_wandb/utils/pickling.py,sha256=8PyzL6l8nnTLOtfKukDSyceoX1888vMLbPIvdUseFME,8332
13
- dagster_wandb-0.25.8.dist-info/LICENSE,sha256=tAkwu8-AdEyGxGoSvJ2gVmQdcicWw3j1ZZueVV74M-E,11357
14
- dagster_wandb-0.25.8.dist-info/METADATA,sha256=L8VYeheMxSTYZ3bFWb7FX2mWvc54x_tIsv9twWuT9L4,842
15
- dagster_wandb-0.25.8.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
16
- dagster_wandb-0.25.8.dist-info/top_level.txt,sha256=eBR6kNspYqHANL4Hy1aXRhq-pYZmPhFg_xKui2AS8cE,14
17
- dagster_wandb-0.25.8.dist-info/RECORD,,
13
+ dagster_wandb-0.25.9.dist-info/LICENSE,sha256=tAkwu8-AdEyGxGoSvJ2gVmQdcicWw3j1ZZueVV74M-E,11357
14
+ dagster_wandb-0.25.9.dist-info/METADATA,sha256=F-jqrsvH4x6f7paU_9mbaicg2ZCAR64-TqqUzRGTyVQ,842
15
+ dagster_wandb-0.25.9.dist-info/WHEEL,sha256=yQN5g4mg4AybRjkgi-9yy4iQEFibGQmlz78Pik5Or-A,92
16
+ dagster_wandb-0.25.9.dist-info/top_level.txt,sha256=eBR6kNspYqHANL4Hy1aXRhq-pYZmPhFg_xKui2AS8cE,14
17
+ dagster_wandb-0.25.9.dist-info/RECORD,,