diff --git a/pipenv/utils/toml.py b/pipenv/utils/toml.py index 5a6b99ecf9..21e89dfed5 100644 --- a/pipenv/utils/toml.py +++ b/pipenv/utils/toml.py @@ -1,5 +1,5 @@ -from pipenv.vendor.toml import decoder, TomlDecoder -from pipenv.vendor import tomlkit +import toml +import tomlkit def cleanup_toml(tml): @@ -42,8 +42,8 @@ def convert_tomlkit_table(section): def convert_toml_table(section): for package, value in section.items(): - if hasattr(value, "keys") and not isinstance(value, decoder.InlineTableDict): - table = TomlDecoder().get_empty_inline_table() + if hasattr(value, "keys") and not isinstance(value, toml.decoder.InlineTableDict): + table = toml.TomlDecoder().get_empty_inline_table() table.update(value) section[package] = table