diff --git a/python_modules/libraries/dagster-wandb/dagster_wandb/io_manager.py b/python_modules/libraries/dagster-wandb/dagster_wandb/io_manager.py index b50090e2879ff..fc7bd8806f259 100644 --- a/python_modules/libraries/dagster-wandb/dagster_wandb/io_manager.py +++ b/python_modules/libraries/dagster-wandb/dagster_wandb/io_manager.py @@ -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 diff --git a/python_modules/libraries/dagster-wandb/dagster_wandb/launch/ops.py b/python_modules/libraries/dagster-wandb/dagster_wandb/launch/ops.py index 382dd8ee98e44..3a0ee67cf80ff 100644 --- a/python_modules/libraries/dagster-wandb/dagster_wandb/launch/ops.py +++ b/python_modules/libraries/dagster-wandb/dagster_wandb/launch/ops.py @@ -1,5 +1,4 @@ from dagster import OpExecutionContext, op - from wandb.sdk.launch import launch from wandb.sdk.launch.launch_add import launch_add diff --git a/python_modules/libraries/dagster-wandb/dagster_wandb/resources.py b/python_modules/libraries/dagster-wandb/dagster_wandb/resources.py index 21edc55ca5d50..ca73d5a3c2bcb 100644 --- a/python_modules/libraries/dagster-wandb/dagster_wandb/resources.py +++ b/python_modules/libraries/dagster-wandb/dagster_wandb/resources.py @@ -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" diff --git a/python_modules/libraries/dagster-wandb/dagster_wandb_tests/test_io_manager.py b/python_modules/libraries/dagster-wandb/dagster_wandb_tests/test_io_manager.py index b0dd0eadfdac2..748b486a17fb3 100644 --- a/python_modules/libraries/dagster-wandb/dagster_wandb_tests/test_io_manager.py +++ b/python_modules/libraries/dagster-wandb/dagster_wandb_tests/test_io_manager.py @@ -2,6 +2,7 @@ from unittest.mock import ANY, MagicMock, patch import pytest +import wandb from callee import EndsWith, Regex from dagster import ( AssetKey, @@ -18,8 +19,6 @@ wandb_artifacts_io_manager, wandb_resource, ) - -import wandb from wandb import Artifact DAGSTER_RUN_ID = "unit-testing"