Great, thanks for letting me know regarding this @Flow . I had tried the first way. On doing pip freeze on the environment getting the current version as 0.17.7 in the databricks notebook. Difficult to perform the 2nd step as this is a production grade code.