mirror of
https://github.com/jupyter/docker-stacks.git
synced 2025-10-07 18:14:05 +00:00
79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
# Copyright (c) Jupyter Development Team.
|
|
# Distributed under the terms of the Modified BSD License.
|
|
from dataclasses import dataclass, field
|
|
from typing import Optional
|
|
|
|
from tagging.manifests import (
|
|
AptPackagesManifest,
|
|
CondaEnvironmentManifest,
|
|
JuliaPackagesManifest,
|
|
ManifestInterface,
|
|
RPackagesManifest,
|
|
SparkInfoManifest,
|
|
)
|
|
from tagging.taggers import (
|
|
DateTagger,
|
|
HadoopVersionTagger,
|
|
JavaVersionTagger,
|
|
JuliaVersionTagger,
|
|
JupyterHubVersionTagger,
|
|
JupyterLabVersionTagger,
|
|
JupyterNotebookVersionTagger,
|
|
PythonVersionTagger,
|
|
RVersionTagger,
|
|
SHATagger,
|
|
SparkVersionTagger,
|
|
TaggerInterface,
|
|
TensorflowVersionTagger,
|
|
UbuntuVersionTagger,
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class ImageDescription:
|
|
parent_image: Optional[str]
|
|
taggers: list[TaggerInterface] = field(default_factory=list)
|
|
manifests: list[ManifestInterface] = field(default_factory=list)
|
|
|
|
|
|
ALL_IMAGES = {
|
|
"base-notebook": ImageDescription(
|
|
parent_image=None,
|
|
taggers=[
|
|
SHATagger(),
|
|
DateTagger(),
|
|
UbuntuVersionTagger(),
|
|
PythonVersionTagger(),
|
|
JupyterNotebookVersionTagger(),
|
|
JupyterLabVersionTagger(),
|
|
JupyterHubVersionTagger(),
|
|
],
|
|
manifests=[CondaEnvironmentManifest(), AptPackagesManifest()],
|
|
),
|
|
"minimal-notebook": ImageDescription(parent_image="base-notebook"),
|
|
"scipy-notebook": ImageDescription(parent_image="minimal-notebook"),
|
|
"r-notebook": ImageDescription(
|
|
parent_image="minimal-notebook",
|
|
taggers=[RVersionTagger()],
|
|
manifests=[RPackagesManifest()],
|
|
),
|
|
"tensorflow-notebook": ImageDescription(
|
|
parent_image="scipy-notebook", taggers=[TensorflowVersionTagger()]
|
|
),
|
|
"datascience-notebook": ImageDescription(
|
|
parent_image="scipy-notebook",
|
|
taggers=[RVersionTagger(), JuliaVersionTagger()],
|
|
manifests=[RPackagesManifest(), JuliaPackagesManifest()],
|
|
),
|
|
"pyspark-notebook": ImageDescription(
|
|
parent_image="scipy-notebook",
|
|
taggers=[SparkVersionTagger(), HadoopVersionTagger(), JavaVersionTagger()],
|
|
manifests=[SparkInfoManifest()],
|
|
),
|
|
"all-spark-notebook": ImageDescription(
|
|
parent_image="pyspark-notebook",
|
|
taggers=[RVersionTagger()],
|
|
manifests=[RPackagesManifest()],
|
|
),
|
|
}
|