def test_from_env(self):
reqs = Python.read_pip_environment(skip_options=True, )
env = Python(requirements=reqs)
assert env._msg.python.requirements
assert not env._msg.python.raw_requirements
reqs = _pip_requirements_utils.pin_verta_and_cloudpickle(reqs)
assert_parsed_reqs_match(env.requirements, reqs)
def test_log_environment(self, registered_model):
model_version = registered_model.get_or_create_version(
name="my version")
reqs = Python.read_pip_environment()
env = Python(requirements=reqs)
model_version.log_environment(env)
model_version = registered_model.get_version(id=model_version.id)
assert str(env) == str(model_version.get_environment())
def test_log_environment(self, registered_model):
model_version = registered_model.get_or_create_version(
name="my version")
reqs = Python.read_pip_environment()
env = Python(requirements=reqs)
model_version.log_environment(env)
model_version = registered_model.get_version(id=model_version.id)
assert str(env) == str(model_version.get_environment())
with pytest.raises(ValueError):
model_version.log_environment(env)
model_version.log_environment(env, overwrite=True)
assert str(env) == str(model_version.get_environment())
def test_del_environment(self, registered_model):
model_version = registered_model.get_or_create_version(
name="my version")
reqs = Python.read_pip_environment()
env = Python(requirements=reqs)
model_version.log_environment(env)
model_version.del_environment()
model_version = registered_model.get_version(id=model_version.id)
assert not model_version.has_environment
with pytest.raises(RuntimeError) as excinfo:
model_version.get_environment()
assert "environment was not previously set" in str(excinfo.value)
def test_skip_spacy_models(self):
pattern = SPACY_MODEL_PATTERN + "=="
requirements = Python.read_pip_environment()
assert not any(re.match(pattern, req) for req in requirements)
def test_reqs_from_env(self):
reqs = Python.read_pip_environment()
env = Python(requirements=reqs)
assert env._msg.python.requirements
