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

Fix tensorflow-version tagging

Browse Source
This commit is contained in:
Ayaz Salikhov
2021-03-22 16:14:53 +03:00
parent 355be0b500
commit 27fe8881a8
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tagging/taggers.py
View File

@@ -20,6 +20,14 @@ def _get_env_variable(container, variable):
raise KeyError(variable) raise KeyError(variable)
def _get_pip_package_version(container, package):
VERSION_PREFIX = "Version: "
package_info = run_simple_command(container, cmd=f"pip show {package}", print_result=False)
version_line = package_info.split("\n")[1]
assert version_line.startswith(VERSION_PREFIX)
return version_line[len(VERSION_PREFIX):]
class TaggerInterface: class TaggerInterface:
"""HooksInterface for all hooks common interface""" """HooksInterface for all hooks common interface"""
@staticmethod @staticmethod
@@ -66,7 +74,7 @@ class RVersionTagger(TaggerInterface):
class TensorflowVersionTagger(TaggerInterface): class TensorflowVersionTagger(TaggerInterface):
@staticmethod @staticmethod
def tag_value(container): def tag_value(container):
return "tensorflow-" + _get_program_version(container, "tensorflow") return "tensorflow-" + _get_pip_package_version(container, "tensorflow")
class JuliaVersionTagger(TaggerInterface): class JuliaVersionTagger(TaggerInterface):