ModuleNotFoundError: No module named 'wandb.sdk.wandb_artifacts' when importing wandb_artifacts_io_manager

[xinbeiG]

Dagster version

dagster, version 1.3.3

What's the issue?

I'm trying to import wandb_artifacts_io_manager like this:

from dagster_wandb import wandb_artifacts_io_manager

Traceback:

  File "/home/Packages/anaconda3/envs/myenv/lib/python3.8/site-packages/dagster_wandb/__init__.py", line 3, in <module>
    from .io_manager import WandbArtifactsIOManagerError, wandb_artifacts_io_manager
  File "/home/Packages/anaconda3/envs/myenv/lib/python3.8/site-packages/dagster_wandb/io_manager.py", line 23, in <module>
    from wandb.sdk.wandb_artifacts import Artifact
ModuleNotFoundError: No module named 'wandb.sdk.wandb_artifacts'

Other package version:

Name: dagster-wandb
Version: 0.19.3

I think the change from wandb/sdk/__init__.py caused this issue.
Changing from wandb.sdk.wandb_artifacts import Artifact to from wandb.sdk.artifacts.artifact import Artifact in dagster_wandb/io_manager.py should be able to fix this issue.

What did you expect to happen?

No response

How to reproduce?

No response

Deployment type

Local

Deployment details

No response

Additional information

No response

Message from the maintainers

Impacted by this issue? Give it a 👍! We factor engagement into prioritization.

[yuhan]

@chrishiste mind taking a look to confirm whether this was intentional or an oversight

[chrishiste]

@xinbeiG Thanks for opening this issue. You are right the change in the SDK broke this import. Fortunately, we already have a PR that includes a fix for this problem as well as other improvements: #15170