def install_transformers(self):
"""Deprecated: Installs the latest version of the Gretel Transformers package
Prefer ``install_packages`` instead.
"""
logger.warning(
"The method, install_transformers is deprecated. Please use install_packages instead."
)
pkg.install_packages(self.api_key, self.host)
def install_packages(self, verbose: bool = False, version: str = "latest"):
"""Installs the latest version of the Gretel Transformers package
Args:
verbose: Will print all package installation messages.
version: Eg: ``1.1.0``. Determines package version to install. Specifying
"latest" will ensure the latest version of the package is installed.
"""
pkg.install_packages(
api_key=self.api_key,
host=self.host,
verbose=verbose,
version=version,
)
def test_does_install_packages(popen, get_package):
popen.return_value = ProcessResponse()
get_package.return_value = OptResponse(status_code=200)
pkg_installers.install_packages("test123", "api-dev.gretel.cloud")
assert popen.call_args[0][0][-4:] == [
"pip",
"--disable-pip-version-check",
"install",
"https://path/to/package.whl",
]
get_package.assert_called_with(
"https://api-dev.gretel.cloud/opt/pkg/gretel-helpers",
headers={"Authorization": "test123"},
params={'version': 'latest'}
)
def test_handles_api_down(get_package):
get_package.return_value = OptResponse(status_code=500)
with pytest.raises(pkg_installers.GretelInstallError):
pkg_installers.install_packages("test123", "api-dev")
def test_requires_api_key():
with pytest.raises(TypeError):
pkg_installers.install_packages() # type: ignore