Skip to content

Commit

Permalink
fix: Ruff
Browse files Browse the repository at this point in the history
  • Loading branch information
chrishiste committed Feb 27, 2024
1 parent b501ce6 commit d3a9df8
Show file tree
Hide file tree
Showing 9 changed files with 10 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,6 @@
io_manager,
)
from dagster._core.storage.io_manager import dagster_maintained_io_manager

from wandb import Artifact
from wandb.data_types import WBValue

Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,4 @@
from dagster import OpExecutionContext, op

from wandb.sdk.launch import launch, launch_add

from .configs import launch_agent_config, launch_config
Expand Down
Original file line number Diff line number Diff line change
@@ -1,9 +1,8 @@
from typing import Any, Dict

import wandb
from dagster import Field, InitResourceContext, String, StringSource, resource
from dagster._core.definitions.resource_definition import dagster_maintained_resource

import wandb
from wandb.sdk.internal.internal_api import Api

WANDB_CLOUD_HOST: str = "https://api.wandb.ai"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

import pytest
from dagster import build_op_context

from dagster_wandb import wandb_resource
from dagster_wandb.launch.ops import raise_on_invalid_config, run_launch_agent, run_launch_job

Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import pytest

from dagster_wandb.utils.errors import (
WandbArtifactsIOManagerError,
raise_on_empty_configuration,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
from unittest.mock import ANY, MagicMock, patch

import pytest
import wandb
from callee import EndsWith, Regex
from dagster import (
AssetKey,
Expand All @@ -13,15 +14,14 @@
build_input_context,
build_output_context,
)
from wandb import Artifact

from dagster_wandb import (
WandbArtifactsIOManagerError,
wandb_artifacts_io_manager,
wandb_resource,
)

import wandb
from wandb import Artifact

DAGSTER_RUN_ID = "unit-testing"
DAGSTER_RUN_ID_SHORT = DAGSTER_RUN_ID[0:8]
DAGSTER_HOME = "/path/to/dagster_home"
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

import pytest
from dagster import build_output_context

from dagster_wandb.utils.pickling import pickle_artifact_content


Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from unittest.mock import patch

from dagster import build_init_resource_context

from dagster_wandb.resources import WANDB_CLOUD_HOST, wandb_resource

API_KEY = "api_key"
Expand Down
2 changes: 2 additions & 0 deletions python_modules/libraries/dagster-wandb/pyproject.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
[tool.ruff]
extend = "../../../pyproject.toml"

0 comments on commit d3a9df8

Please sign in to comment.