mirror of
https://github.com/jupyter/docker-stacks.git
synced 2025-10-08 10:34:06 +00:00
278dd76837
* Add PyTorch image * Fix linting errors * Fix link to pytorch website * Address review comments * Fix PytorchVersionTagger * Remove "+cpu" suffix from pytorch version tag * Update selecting.md * Rename pytorch-notebook/.dockerignore to images/pytorch-notebook/.dockerignore * Rename pytorch-notebook/Dockerfile to images/pytorch-notebook/Dockerfile * Rename pytorch-notebook/README.md to images/pytorch-notebook/README.md * Add pytorch-notebook to registry-overviews * Add registry to pytorch image * Use Quay.io * Remove incorrect link * Update action.yml * Update docker.yml * Remove information about Docker Hub, because this image won't be uploaded to Docker Hub * Update docker.yml * Update action.yml * Add pytorch-notebook to registry-move.yml --------- Co-authored-by: Ayaz Salikhov <mathbunnyru@users.noreply.github.com> Co-authored-by: Ayaz Salikhov <mathbunnyru@gmail.com>
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
# Copyright (c) Jupyter Development Team.
|
|
# Distributed under the terms of the Modified BSD License.
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
THIS_DIR = Path(__file__).parent.resolve()
|
|
|
|
# Please, take a look at the hierarchy of the images here:
|
|
# https://jupyter-docker-stacks.readthedocs.io/en/latest/using/selecting.html#image-relationships
|
|
ALL_IMAGES = {
|
|
"docker-stacks-foundation": None,
|
|
"base-notebook": "docker-stacks-foundation",
|
|
"minimal-notebook": "base-notebook",
|
|
"scipy-notebook": "minimal-notebook",
|
|
"r-notebook": "minimal-notebook",
|
|
"julia-notebook": "minimal-notebook",
|
|
"tensorflow-notebook": "scipy-notebook",
|
|
"pytorch-notebook": "scipy-notebook",
|
|
"datascience-notebook": "scipy-notebook",
|
|
"pyspark-notebook": "scipy-notebook",
|
|
"all-spark-notebook": "pyspark-notebook",
|
|
}
|
|
|
|
|
|
def get_test_dirs(
|
|
short_image_name: Optional[str],
|
|
) -> list[Path]:
|
|
if short_image_name is None:
|
|
return []
|
|
|
|
test_dirs = get_test_dirs(ALL_IMAGES[short_image_name])
|
|
if (current_image_tests_dir := THIS_DIR / short_image_name).exists():
|
|
test_dirs.append(current_image_tests_dir)
|
|
return test_dirs
|