diff --git a/tagging/images_hierarchy.py b/tagging/images_hierarchy.py index 68f837fd..145b329a 100644 --- a/tagging/images_hierarchy.py +++ b/tagging/images_hierarchy.py @@ -19,6 +19,7 @@ from tagging.taggers import ( JupyterHubVersionTagger, JupyterLabVersionTagger, JupyterNotebookVersionTagger, + PythonMajorMinorVersionTagger, PythonVersionTagger, RVersionTagger, SHATagger, @@ -43,6 +44,7 @@ ALL_IMAGES = { SHATagger(), DateTagger(), UbuntuVersionTagger(), + PythonMajorMinorVersionTagger(), PythonVersionTagger(), JupyterNotebookVersionTagger(), JupyterLabVersionTagger(), diff --git a/tagging/taggers.py b/tagging/taggers.py index de896f5f..85537117 100644 --- a/tagging/taggers.py +++ b/tagging/taggers.py @@ -76,6 +76,13 @@ class PythonVersionTagger(TaggerInterface): 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(".")] + + class JupyterNotebookVersionTagger(TaggerInterface): @staticmethod def tag_value(container: Container) -> str: