From 9d6cbe97888f26adef76657dc39600f84a2fe51b Mon Sep 17 00:00:00 2001
From: Kuba <your.email@example.com>
Date: Tue, 20 Feb 2024 14:50:11 +0100
Subject: [PATCH] Fix imports

---
 .../api/templates/models/cisco_system.py      |  2 +-
 .../models/cisco_vpn_interface_model.py       |  2 +-
 .../tests/templates/models/cisco_vpn.py       | 22 -------------------
 3 files changed, 2 insertions(+), 24 deletions(-)

diff --git a/catalystwan/api/templates/models/cisco_system.py b/catalystwan/api/templates/models/cisco_system.py
index 4efea8bc7..f5db35e64 100644
--- a/catalystwan/api/templates/models/cisco_system.py
+++ b/catalystwan/api/templates/models/cisco_system.py
@@ -1,8 +1,8 @@
 from pathlib import Path
 from typing import ClassVar, List, Literal, Optional
-from typing_extensions import Annotated
 
 from pydantic import BaseModel, BeforeValidator, ConfigDict, Field
+from typing_extensions import Annotated
 
 from catalystwan.api.templates.device_variable import DeviceVariable
 from catalystwan.api.templates.feature_template import FeatureTemplate
diff --git a/catalystwan/api/templates/models/cisco_vpn_interface_model.py b/catalystwan/api/templates/models/cisco_vpn_interface_model.py
index 174f0b641..286b16849 100644
--- a/catalystwan/api/templates/models/cisco_vpn_interface_model.py
+++ b/catalystwan/api/templates/models/cisco_vpn_interface_model.py
@@ -2,9 +2,9 @@
 from enum import Enum
 from pathlib import Path
 from typing import ClassVar, List, Literal, Optional, Union
-from typing_extensions import Annotated
 
 from pydantic import BeforeValidator, ConfigDict, Field
+from typing_extensions import Annotated
 
 from catalystwan.api.templates.feature_template import FeatureTemplate
 from catalystwan.api.templates.models.cisco_omp_model import DEFAULT_OMP_SENDPATH_LIMIT
diff --git a/catalystwan/tests/templates/models/cisco_vpn.py b/catalystwan/tests/templates/models/cisco_vpn.py
index 3c11f966c..fefc84d54 100644
--- a/catalystwan/tests/templates/models/cisco_vpn.py
+++ b/catalystwan/tests/templates/models/cisco_vpn.py
@@ -2,53 +2,31 @@
 
 from catalystwan.api.templates.models.cisco_vpn_model import (
     Advertise,
-    AdvertiseProtocol,
-    AdvertiseProtocolSubType,
     CiscoVPNModel,
-    Direction,
     Dns,
     DnsIpv6,
     GreRoute,
     Host,
     IpsecRoute,
     Ipv6Advertise,
-    Ipv6AdvertiseProtocol,
-    Ipv6AdvertiseProtocolSubType,
-    LeakFromGlobalProtocol,
-    Nat,
     Natpool,
     NextHop,
     NextHopv6,
-    Overload,
     Pool,
     PortForward,
     PrefixList,
-    Proto,
-    Region,
-    Role,
     RouteExport,
-    RouteExportProtocol,
-    RouteExportProtocolSubType,
     RouteExportRedistribute,
-    RouteExportRedistributeProtocol,
     RouteImport,
     RouteImportFrom,
-    RouteImportFromProtocol,
-    RouteImportFromProtocolSubType,
     RouteImportFromRedistribute,
-    RouteImportFromRedistributeProtocol,
-    RouteImportProtocol,
-    RouteImportProtocolSubType,
     RouteImportRedistribute,
-    RouteImportRedistributeProtocol,
     Routev4,
     Routev6,
     Service,
     ServiceRoute,
     Static,
-    StaticNatDirection,
     SubnetStatic,
-    SvcType,
 )
 from catalystwan.utils.device_model import DeviceModel