datarobot-community / custom-models

Various DataRobot MLOps examples
Apache License 2.0
26 stars 56 forks source link

Solution to problems with agent for version > 8 #12

Open ToonWeyens opened 2 years ago

ToonWeyens commented 2 years ago

Thanks @timsetsfire for figuring this out: Make sure you have the same version number for both datarobot-mlops and datarobot-mlops-connected-client.

This means that you either install both from pypi with pip, or you both install them from developer tools, with something similar to

## grab mlops agents tarball
mlops_agents_tb = client.get("mlopsInstaller") 
with open("/content/mlops-agent.tar.gz", "wb") as f:
    f.write(mlops_agents_tb.content)
## unpack tarball
os.system("tar -xf /content/mlops-agent.tar.gz -C . && mkdir -p /tmp/ta")

import glob
agents_dir = glob.glob("/content/datarobot_mlops*").pop()

os.system('pip install {}/lib/datarobot_mlops-*.whl'.format(agents_dir))
os.system('pip install {}/lib/datarobot_mlops_connected_client-*.whl'.format(agents_dir))

This is obvious in hindsight but multiple of our notebooks had problems with this :-)