def test_create_gist_simple_with_gist_slug(self, mock_requests_post):
with fake_creds('.jovian', 'credentials.json'):
create_gist_simple(
filename=
'jovian/tests/resources/creds/.jovian/credentials.json',
gist_slug='fake_gist_slug',
title='Credentials',
version_title='first version')
mock_requests_post.assert_called_with(
'https://api-staging.jovian.ai/gist/fake_gist_slug/upload',
data={'version_title': 'first version'},
files={
'files':
('jovian/tests/resources/creds/.jovian/credentials.json',
ANY)
},
headers={
"Authorization": "Bearer fake_api_key",
"x-jovian-source": "library",
"x-jovian-library-version": __version__,
"x-jovian-guest": "b6538d4dfde04fcf993463a828a9cec6",
"x-jovian-org": "staging"
},
json=None)
def test_create_gist_simple_with_gist_slug(mock_requests_post):
with fake_creds() as dir:
create_gist_simple(
filename=os.path.join(dir, ".jovian/credentials.json"),
gist_slug="fake_gist_slug",
title="Credentials",
version_title="first version",
)
mock_requests_post.assert_called_with(
"https://api-staging.jovian.ai/gist/fake_gist_slug/upload",
data={"version_title": "first version"},
files={
"files": (
os.path.join(dir, ".jovian/credentials.json"),
ANY,
)
},
headers={
"Authorization": "Bearer fake_api_key",
"x-jovian-source": "library",
"x-jovian-library-version": __version__,
"x-jovian-guest": "b6538d4dfde04fcf993463a828a9cec6",
"x-jovian-org": "staging",
},
json=None,
)
def test_create_gist_simple_has_warning(mock_post_request, capsys):
with fake_creds() as dir:
create_gist_simple(
filename=os.path.join(dir, ".jovian/credentials.json"))
assert capsys.readouterr().err.strip(
) == '[jovian] Error: Uploaded gist has a warning'.strip()
def test_create_gist_simple_raises_api_error(mock_requests_post):
with fake_creds() as dir:
# setUp
creds = {
"WEBAPP_URL": "https://staging.jovian.ml/",
"GUEST_KEY": "b6538d4dfde04fcf993463a828a9cec6",
"API_URL": "https://api-staging.jovian.ai",
"API_KEY": "fake_invalid_api_key",
"ORG_ID": "staging",
}
write_creds(creds)
with pytest.raises(ApiError) as context:
create_gist_simple(
filename=os.path.join(dir, ".jovian/credentials.json"),
title="Credentials",
version_title="first version",
)
mock_requests_post.assert_called_with(
"https://api-staging.jovian.ai/gist/create",
data={
"visibility": "auto",
"public": True,
"title": "Credentials",
"version_title": "first version",
},
files={
"files": (
os.path.join(dir, ".jovian/credentials.json"),
ANY,
)
},
headers={
"Authorization": "Bearer fake_invalid_api_key",
"x-jovian-source": "library",
"x-jovian-library-version": __version__,
"x-jovian-guest": "b6538d4dfde04fcf993463a828a9cec6",
"x-jovian-org": "staging",
},
json=None,
)
assert (
str(context.value)
== "File upload failed: (HTTP 404) Gist not found"
)
def test_create_gist_simple_raises_api_error(self, mock_requests_post):
with fake_creds('.jovian', 'credentials.json'):
# setUp
creds = {
"WEBAPP_URL": "https://staging.jovian.ml/",
"GUEST_KEY": "b6538d4dfde04fcf993463a828a9cec6",
"API_URL": "https://api-staging.jovian.ai",
"API_KEY": "fake_invalid_api_key",
"ORG_ID": "staging"
}
write_creds(creds)
with self.assertRaises(ApiError) as context:
create_gist_simple(
filename=
'jovian/tests/resources/creds/.jovian/credentials.json',
title='Credentials',
version_title='first version')
mock_requests_post.assert_called_with(
'https://api-staging.jovian.ai/gist/create',
data={
'visibility': 'auto',
'public': True,
'title': 'Credentials',
'version_title': 'first version'
},
files={
'files':
('jovian/tests/resources/creds/.jovian/credentials.json',
ANY)
},
headers={
"Authorization": "Bearer fake_invalid_api_key",
"x-jovian-source": "library",
"x-jovian-library-version": __version__,
"x-jovian-guest": "b6538d4dfde04fcf993463a828a9cec6",
"x-jovian-org": "staging"
},
json=None)
assert context.exception.args[
0] == 'File upload failed: (HTTP 404) Gist not found'
def commit(message=None,
files=[],
outputs=[],
environment='auto',
privacy='auto',
filename=None,
project=None,
new_project=None,
git_commit=False,
git_message='auto',
**kwargs):
"""Uploads the current file (Jupyter notebook or python script) to |Jovian|
Saves the checkpoint of the notebook, captures the required dependencies from
the python environment and uploads the notebook, env file, additional files like scripts, csv etc.
to |Jovian|. Capturing the python environment ensures that the notebook can be reproduced.
Args:
message(string, optional): A short message to be used as the title for this version.
files(array, optional): Any additional scripts(.py files), CSVs etc. that are required to
run the notebook. These will be available in the files tab of the project page on Jovian.ml
outputs(array, optional): Any outputs files or artifacts generated from the modeling processing.
This can include model weights/checkpoints, generated CSVs, output images etc.
environment(string, optional): The type of Python environment to be captured. Allowed options are
'conda' , 'pip', 'auto' (for automatic detection) and None (to skip environment capture).
privacy(bool, optional): Privacy level of the project (if a new one is being created).
* 'auto' - use account level settings. Defaults to 'public'
* 'public' - visible on profile and publicly accessible/searchable
* 'secret' - not on profile only accessible via the direct link
* 'private' - only for the accessible to owner and collaborators
This argument has no effect on existing project. Change the privacy settings of a existing notebook
on the webapp.
filename(string, optional): The filename of the current Jupyter notebook or Python script. This is
detected automatically in most cases, but in certain environments like Jupyter Lab or password protected notebooks, the detection
may fail and the filename needs to be provided using this argument.
project(string, optional): Name of the |Jovian| project to which the current notebook/file should
be committed. Format: 'username/title' e.g. 'aakashns/jovian-example' or 'jovian-example'
(username of current user inferred automatically). If the project does not exist, a new one is
created. If it exists, the current notebook is added as a new version to the existing project, if
you are a owner/collaborator. If left empty, project name is picked up from the `.jovianrc` file in the
current directory, or a new project is created using the filename as the project name.
new_project(bool, optional): Whether to create a new project or update the existing one. Allowed option
are False (use the existing project, if a .jovianrc file exists, if available), True (create a new project)
git_commit(bool, optional): If True, also performs a Git commit and records the commit hash. This is
applicable only when the notebook is inside a Git repository.
git_message(string, optional): Commit message for git. If not provided, it uses the `message` argument
.. attention::
Pass notebook's name to `filename` argument, in certain environments like Jupyter Lab and password protected
notebooks sometimes it may fail to detect notebook automatically.
.. |Jovian| raw:: html
<a href="https://jovian.ml/?utm_source=docs" target="_blank"> Jovian.ml </a>
"""
global _current_slug
# Deprecated argument (secret)
if privacy == 'auto' and 'secret' in kwargs:
privacy = 'secret' if kwargs['secret'] else 'auto'
log('"secret" is deprecated. Use "privacy" instead (allowed options: ' +
'"public", "private", "secret", "auto")', error=True)
# Deprecated argument (nb_filename)
if filename is None and 'nb_filename' in kwargs:
filename = kwargs['nb_filename']
log('"nb_filename" is deprecated. Use "filename" instead', error=True)
# Deprecated argument (env_type)
if 'env_type' in kwargs:
environment = kwargs['environment']
log('"env_type" is deprecated. Use "environment" instead', error=True)
# Deprecated argument (capture_env)
if 'capture_env' in kwargs and not kwargs['capture_env']:
environment = None
log('"catpure_env" is deprecated. Use "environment=None" instead', error=True)
# Deprecated argument (notebook_id)
if 'notebook_id' in kwargs:
project = kwargs['notebook_id']
log('"notebook_id" is deprecated. Use "project" instead.', error=True)
# Deprecated argument (create_new)
if 'create_new' in kwargs:
new_project = kwargs['create_new']
log('"create_new" is deprecated. Use "new_project" instead.', error=True)
# Deprecated argument (artifacts)
if 'artifacts' in kwargs:
outputs = kwargs['artifacts']
log('"artifacts" is deprecated. Use "outputs" instead', error=True)
# Skip if unsupported environment
if not in_script() and not in_notebook():
log('Failed to detect Jupyter notebook or Python script. Skipping..', error=True)
return
# Attempt to save Jupyter notebook
if in_notebook():
save_notebook()
log('Attempting to save notebook..')
sleep(1)
# Extract notebook/script filename
filename = _parse_filename(filename)
if filename is None:
log(FILENAME_MSG)
return
# Ensure that the file exists
if not os.path.exists(filename):
log('The detected/provided file "' + filename +
'" does not exist. Please provide the correct notebook filename ' +
'as the "filename" argument to "jovian.commit".')
return
# Retrieve Gist ID & title
project_title, project_id = _parse_project(project, filename, new_project)
# Create or update gist (with title and )
res = api.create_gist_simple(filename, project_id, privacy, project_title, message)
slug, owner, version, title = res['slug'], res['owner'], res['version'], res['title']
username = owner['username']
# Cache slug for further commits
_current_slug = slug
set_notebook_slug(filename, slug)
# Attach environment, files and outputs
_capture_environment(environment, slug, version)
_attach_files(files, slug, version)
_attach_files(outputs, slug, version, output=True)
if not git_message or git_message == 'auto':
git_message = message or 'jovian commit ' + username + '/' + title + ' v' + str(version)
_perform_git_commit(filename, git_commit, git_message)
_attach_records(slug, version)
log('Committed successfully! ' + read_webapp_url() + username + "/" + title)
def commit(secret=False,
nb_filename=None,
files=[],
capture_env=True,
env_type='conda',
notebook_id=None,
create_new=None):
"""Save the notebook, capture environment and upload to the cloud for sharing.
In most cases, commit works well with the default arguments. It attempts to
1. Save the Jupyter notebook
2. Upload the notebook (and additional scripts, CSVs etc.) to https://jvn.io
3. Capture the python environment (using Anaconda or pip)
4. Upload the python environment to cloud
Capturing the python environment ensures that the notebook can be reproduced and
executed easily using the `jovian` command line tool. For more details, see
https://jvn.io/getting-started .
Issues and bugs can be reported here: https://github.com/swiftace-ai/jovian-py
Arguments:
secret (bool, optional): Create a secret notebook on https://jvn.io , which is only
accessible via the link, and is not visible on the owner's public profile. By default,
commited notebooks are public and visible on the owner's profile.
nb_filename (string, optional): The filename of the jupyter notebook (including
the .ipynb extension). This is detected automatically in most cases, but in
certain environments like Jupyter Lab, the detection may fail and the filename
needs to be provided using this argument.
files (array, optional): Any additional scripts (.py files), CSVs that are required to
run the notebook. These will be available in the files tab on https://jvn.io .
capture_env (bool, optional): If `True`, the Python environment (python version,
libraries etc.) are captured and uploaded along with the notebook.
env_type (string, optional): The type of environment to be captured. Allowed options are
'conda' and 'pip'.
notebook_id (string, optional): If you wish to update an existing notebook owned by you,
you can use this argument to provide the base64 ID (present in the URL) of an notebook
hosted on https://jvn.io . In most cases, this argument is not required, and the library
can automatically infer whether you are looking to update an existing notebook or create
a new one.
create_new (bool, optional): If set to True, doesn't update the existing notebook on
https://jvn.io (if one is detected). Instead, it creates a new notebook when commit is called.
"""
global _current_slug
# Check if we're in a Jupyter environment
if not in_notebook():
log('Failed to detect Juptyer notebook. Skipping..', error=True)
return
# Save the notebook (uses Javascript, doesn't work everywhere)
log('Saving notebook..')
save_notebook()
sleep(1)
# Get the filename of the notebook (if not provided)
if nb_filename is None:
nb_filename = get_notebook_name()
# Exit with help message if filename wasn't detected (or provided)
if nb_filename is None:
log(FILENAME_MSG)
return
# Check whether to create a new gist, or update an old one
if not create_new and notebook_id is None:
# First preference to the in-memory slug variable
if _current_slug is not None:
notebook_id = _current_slug
else:
notebook_id = get_notebook_slug(nb_filename)
# Check if the current user can push to this slug
if notebook_id is not None:
gist_meta = get_gist(notebook_id)
if not gist_meta['isOwner']:
notebook_id = None
# Log whether this is an update or creation
if notebook_id is None:
log('Creating a new notebook on https://jvn.io')
else:
log('Updating notebook "' + notebook_id + '" on https://jvn.io')
# Upload the notebook & create/update the gist
res = create_gist_simple(nb_filename, notebook_id, secret)
if res is None:
return
# Extract slug and owner from created gist
slug, owner = res['slug'], res['owner']
# Set/update the slug information
_current_slug = slug
set_notebook_slug(nb_filename, slug)
# Save & upload environment
if capture_env:
log('Capturing environment..')
if env_type == 'conda':
# Capture conda environment
try:
upload_conda_env(slug)
except CondaError as e:
log(str(e), error=True)
elif env_type == 'pip':
# Capture pip environment
try:
upload_pip_env(slug)
except Exception as e:
log(str(e), error=True)
# Upload additional files
if files and len(files) > 0:
log('Uploading additional files..')
# Upload each file
for fname in files:
if os.path.exists(fname) and not os.path.isdir(fname):
try:
upload_file(slug, fname)
except Exception as e:
log(str(e), error=True)
elif os.path.isdir(fname):
log('Ignoring directory "' + fname + '"', error=True)
else:
log('Ignoring "' + fname + '" (not found)', error=True)
# Print commit URL
log('Committed successfully! ' + WEBAPP_URL + "/" + owner['username'] +
"/" + slug)
