hazza/docker-stacks
1
0
Fork 0
mirror of https://github.com/jupyter/docker-stacks.git synced 2025-10-18 07:22:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make taggers manifest functions (#2252)

Browse Source 
* Make taggers and manifests functions

* Add changelog
This commit is contained in:
Ayaz Salikhov
2025-03-12 16:05:09 +00:00
committed by GitHub
parent 48b065050c
commit 2c1df4020a
17 changed files with 183 additions and 253 deletions
Download Patch File Download Diff File Expand all files Collapse all files

101
tagging/taggers/versions.py
View File

@@ -2,7 +2,6 @@
# Distributed under the terms of the Modified BSD License.
from docker.models.containers import Container
from tagging.taggers.tagger_interface import TaggerInterface
from tagging.utils.docker_runner import DockerRunner
@@ -23,92 +22,66 @@ def _get_pip_package_version(container: Container, package: str) -> str:
return version_line[len(PIP_VERSION_PREFIX) :]
class PythonVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
return "python-" + _get_program_version(container, "python").split()[1]
def python_tagger(container: Container) -> str:
return "python-" + _get_program_version(container, "python").split()[1]
class PythonMajorMinorVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
full_version = PythonVersionTagger.tag_value(container)
return full_version[: full_version.rfind(".")]
def python_major_minor_tagger(container: Container) -> str:
full_version = python_tagger(container)
return full_version[: full_version.rfind(".")]
class MambaVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
return "mamba-" + _get_program_version(container, "mamba")
def mamba_tagger(container: Container) -> str:
return "mamba-" + _get_program_version(container, "mamba")
class CondaVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
return "conda-" + _get_program_version(container, "conda").split()[1]
def conda_tagger(container: Container) -> str:
return "conda-" + _get_program_version(container, "conda").split()[1]
class JupyterNotebookVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
return "notebook-" + _get_program_version(container, "jupyter-notebook")
def jupyter_notebook_tagger(container: Container) -> str:
return "notebook-" + _get_program_version(container, "jupyter-notebook")
class JupyterLabVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
return "lab-" + _get_program_version(container, "jupyter-lab")
def jupyter_lab_tagger(container: Container) -> str:
return "lab-" + _get_program_version(container, "jupyter-lab")
class JupyterHubVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
return "hub-" + _get_program_version(container, "jupyterhub")
def jupyter_hub_tagger(container: Container) -> str:
return "hub-" + _get_program_version(container, "jupyterhub")
class RVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
return "r-" + _get_program_version(container, "R").split()[2]
def r_tagger(container: Container) -> str:
return "r-" + _get_program_version(container, "R").split()[2]
class JuliaVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
return "julia-" + _get_program_version(container, "julia").split()[2]
def julia_tagger(container: Container) -> str:
return "julia-" + _get_program_version(container, "julia").split()[2]
class TensorflowVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
try:
return "tensorflow-" + _get_pip_package_version(container, "tensorflow")
except AssertionError:
return "tensorflow-" + _get_pip_package_version(container, "tensorflow-cpu")
def tensorflow_tagger(container: Container) -> str:
try:
return "tensorflow-" + _get_pip_package_version(container, "tensorflow")
except AssertionError:
return "tensorflow-" + _get_pip_package_version(container, "tensorflow-cpu")
class PytorchVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
return "pytorch-" + _get_pip_package_version(container, "torch").split("+")[0]
def pytorch_tagger(container: Container) -> str:
return "pytorch-" + _get_pip_package_version(container, "torch").split("+")[0]
class SparkVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
SPARK_VERSION_LINE_PREFIX = r" /___/ .__/\_,_/_/ /_/\_\ version"
def spark_tagger(container: Container) -> str:
SPARK_VERSION_LINE_PREFIX = r" /___/ .__/\_,_/_/ /_/\_\ version"
spark_version = _get_program_version(container, "spark-submit")
version_line = next(
filter(
lambda line: line.startswith(SPARK_VERSION_LINE_PREFIX),
spark_version.split("\n"),
)
spark_version = _get_program_version(container, "spark-submit")
version_line = next(
filter(
lambda line: line.startswith(SPARK_VERSION_LINE_PREFIX),
spark_version.split("\n"),
)
return "spark-" + version_line.split(" ")[-1]
)
return "spark-" + version_line.split(" ")[-1]
class JavaVersionTagger(TaggerInterface):
@staticmethod
def tag_value(container: Container) -> str:
return "java-" + _get_program_version(container, "java").split()[1]
def java_tagger(container: Container) -> str:
return "java-" + _get_program_version(container, "java").split()[1]