dagster-wandb 0.25.8__tar.gz → 0.25.9__tar.gz
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.
- {dagster-wandb-0.25.8/dagster_wandb.egg-info → dagster-wandb-0.25.9}/PKG-INFO +1 -1
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb/io_manager.py +2 -8
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb/resources.py +2 -2
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb/types.py +7 -14
- dagster-wandb-0.25.9/dagster_wandb/version.py +1 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9/dagster_wandb.egg-info}/PKG-INFO +1 -1
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb.egg-info/requires.txt +1 -1
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/setup.py +2 -3
- dagster-wandb-0.25.8/dagster_wandb/version.py +0 -1
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/LICENSE +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/MANIFEST.in +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/README.md +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb/__init__.py +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb/launch/__init__.py +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb/launch/configs.py +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb/launch/ops.py +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb/py.typed +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb/utils/__init__.py +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb/utils/errors.py +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb/utils/pickling.py +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb.egg-info/SOURCES.txt +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb.egg-info/dependency_links.txt +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb.egg-info/not-zip-safe +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/dagster_wandb.egg-info/top_level.txt +0 -0
- {dagster-wandb-0.25.8 → dagster-wandb-0.25.9}/setup.cfg +0 -0
|
@@ -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
|
|
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[
|
|
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
|
|
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) ->
|
|
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.
|
|
@@ -1,18 +1,11 @@
|
|
|
1
|
-
import
|
|
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:
|
|
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:
|
|
25
|
-
add_dirs:
|
|
26
|
-
add_files:
|
|
27
|
-
add_references:
|
|
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:
|
|
22
|
+
partitions: dict[str, dict[str, Any]]
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
__version__ = "0.25.9"
|
|
@@ -1,11 +1,10 @@
|
|
|
1
1
|
from pathlib import Path
|
|
2
|
-
from typing import Dict
|
|
3
2
|
|
|
4
3
|
from setuptools import find_packages, setup
|
|
5
4
|
|
|
6
5
|
|
|
7
6
|
def get_version() -> str:
|
|
8
|
-
version:
|
|
7
|
+
version: dict[str, str] = {}
|
|
9
8
|
with open(Path(__file__).parent / "dagster_wandb/version.py", encoding="utf8") as fp:
|
|
10
9
|
exec(fp.read(), version)
|
|
11
10
|
|
|
@@ -34,7 +33,7 @@ setup(
|
|
|
34
33
|
include_package_data=True,
|
|
35
34
|
python_requires=">=3.9,<3.13",
|
|
36
35
|
install_requires=[
|
|
37
|
-
"dagster==1.9.
|
|
36
|
+
"dagster==1.9.9",
|
|
38
37
|
"wandb>=0.15.11,<1.0",
|
|
39
38
|
],
|
|
40
39
|
extras_require={"dev": ["cloudpickle", "joblib", "callee", "dill"]},
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
__version__ = "0.25.8"
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|