[BUG][SCRIPT]Artifact download not working in script mode : ERROR 'graphql: got null for non-null'

wandb --version && python --version && uname

• Weights and Biases version: wandb, version 0.9.1
• Python version: Python 3.6.9 :: Anaconda, Inc.
• Operating System: Linux

Description

Unable to download artifacts using Python3 script while it’s working on jupyter notebook

What I Did

This code works in jupyter notebook

import wandb
run = wandb.init()

artifact = run.use_artifact('jilijeanlouis/ovh-automl-image-classification-train-resnet50-homey-brown-dhole/export-models:v1', type='model')
artifact_dir = artifact.download()

But in python script I get:

>>> artifact = run.use_artifact('jilijeanlouis/ovh-automl-image-classification-train-resnet50-homey-brown-dhole/export-models:v1', type='model')
Traceback (most recent call last):
>>>   File "<stdin>", line 1, in <module>
  File "/opt/conda/lib/python3.6/site-packages/wandb/wandb_run.py", line 430, in use_artifact
    self.api.use_artifact(artifact.id)
  File "/opt/conda/lib/python3.6/site-packages/wandb/apis/internal.py", line 1038, in use_artifact
    'artifactID': artifact_id,
  File "/opt/conda/lib/python3.6/site-packages/wandb/retry.py", line 95, in __call__
    result = self._call_fn(*args, **kwargs)
  File "/opt/conda/lib/python3.6/site-packages/wandb/apis/internal.py", line 108, in execute
    return self.client.execute(*args, **kwargs)
  File "/opt/conda/lib/python3.6/site-packages/gql/client.py", line 54, in execute
    raise Exception(str(result.errors[0]))
Exception: {'message': 'graphql: got null for non-null'}

[vanpelt]

Nice @jqueguiner we’ve fixed this issue in master. We’re targeting a new release first thing next week!

[issue-label-bot[bot]]

Issue-Label Bot is automatically applying the label bug to this issue, with a confidence of 0.94. Please mark this comment with 👍 or 👎 to give our bot feedback!

Links: app homepage, dashboard and code for this bot.