ci: More ci fixes

nautobot_design_builder/contrib/tests/test_ext.py

@@ -301,9 +301,10 @@ class TestBGPExtension(TestCase):
     """Test BGP extension."""
 
     def setUp(self):
-        # TODO: Remove this when BGP models is migrated to 2.0
-        if nautobot_version >= "2.0.0":
-            self.skipTest("BGP Models is not supported in Nautobot 2.x")
+        try:
+            import nautobot_bgp_models.models  # pylint:disable=import-outside-toplevel,unused-import,import-error # noqa: F401
+        except ModuleNotFoundError:
+            self.skipTest("BGP Models app is not installed, skipping associated tests")
         super().setUp()
 
     def test_creation(self):

pyproject.toml

@@ -29,6 +29,7 @@ packages = [
 python = ">=3.8,<3.12"
 # Used for local development
 nautobot = ">=1.6.0,<=2.9999"
+nautobot-bgp-models = { version = "*", optional = true }
 
 [tool.poetry.group.dev.dependencies]
 bandit = "*"
@@ -46,8 +47,6 @@ yamllint = "*"
 toml = "*"
 Markdown = "*"
 
-nautobot-bgp-models = "*"
-
 # Rendering docs to HTML
 mkdocs = "1.5.2"
 # Material for MkDocs theme
@@ -60,6 +59,7 @@ mkdocstrings-python = "1.5.2"
 
 [tool.poetry.extras]
 nautobot = ["nautobot"]
+contrib = ["nautobot-bgp-models"]
 
 [tool.black]
 line-length = 120