fix: Support parsing inline-table format of Poetry dependency spec

nautobot · May 2, 2024 · 94e3e58 · 94e3e58
1 parent 40eec08
commit 94e3e58
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tasks.py b/tasks.py
@@ -44,7 +44,10 @@ def is_truthy(arg):
 with open("pyproject.toml", "r", encoding="utf8") as pyproject:
     parsed_toml = toml.load(pyproject)
 
-NAUTOBOT_VERSION = parsed_toml["tool"]["poetry"]["dependencies"]["nautobot"]
+try:
+    NAUTOBOT_VERSION = parsed_toml["tool"]["poetry"]["dependencies"]["nautobot"]["version"]
+except TypeError:
+    NAUTOBOT_VERSION = parsed_toml["tool"]["poetry"]["dependencies"]["nautobot"]
 
 
 def task(function=None, *args, **kwargs):  # pylint: disable=keyword-arg-before-vararg