fixing typing import

OpenFreeEnergy · Nov 14, 2024 · aff2dfb · aff2dfb
1 parent e568ccf
commit aff2dfb
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/gufe/network.py b/gufe/network.py
@@ -2,18 +2,15 @@
 # For details, see https://github.com/OpenFreeEnergy/gufe
 
 from typing import Generator, Iterable, Optional
+from typing_extensions import Self # Self is included in typing as of python 3.11
 
 import networkx as nx
 from .tokenization import GufeTokenizable
 
 from .chemicalsystem import ChemicalSystem
 from .transformations import Transformation
 
-# Self typing is supported in python 3.11+
-try:
-    from typing import Self
-except ImportError:
-    from __future__ import annotations
+
 
 class AlchemicalNetwork(GufeTokenizable):
     _edges: frozenset[Transformation]