tvalentyn commented on a change in pull request #12115:
URL: https://github.com/apache/beam/pull/12115#discussion_r452456541
##########
File path: sdks/python/container/boot.go
##########
@@ -170,6 +176,28 @@ func main() {
log.Fatalf("Python exited: %v", execx.Execute("python", args...))
}
+// setup wheel specs according to installed python version
+func setupAcceptableWheelSpecs() error {
+ cmd := exec.Command("python", "-V")
+ stdoutStderr, err := cmd.CombinedOutput()
+ if err != nil {
+ return err
+ }
+ re := regexp.MustCompile(`Python (\d)\.(\d).*`)
+ pyVersions := re.FindStringSubmatch(string(stdoutStderr[:]))
+ if len(pyVersions) != 3 {
+ return fmt.Errorf("cannot get parse Python version from %s",
stdoutStderr)
+ }
+ pyVersion := fmt.Sprintf("%s%s", pyVersions[1], pyVersions[2])
+ if pyVersion == "27" {
+ acceptableWhlSpecs = append(acceptableWhlSpecs,
"cp27-cp27mu-manylinux1_x86_64.whl")
+ } else {
+ wheelName := fmt.Sprintf("cp%s-cp%sm-manylinux1_x86_64.whl",
pyVersion, pyVersion)
Review comment:
Thanks! Also filed https://issues.apache.org/jira/browse/BEAM-10434 to
fix the stager.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
