Change PurePath to Path

OpenFreeEnergy · Dec 13, 2023 · 64109d8 · 64109d8
1 parent 4704336
commit 64109d8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/openfe/protocols/openmm_md/plain_md_methods.py b/openfe/protocols/openmm_md/plain_md_methods.py
@@ -72,7 +72,7 @@ def get_uncertainty(self):
 
         return None
 
-    def get_traj_filename(self) -> list[pathlib.PurePath]:
+    def get_traj_filename(self) -> list[pathlib.Path]:
         """
         Get a list of trajectory paths
 
@@ -85,14 +85,14 @@ def get_traj_filename(self) -> list[pathlib.PurePath]:
 
         return traj
 
-    def get_pdb_filename(self) -> list[pathlib.PurePath]:
+    def get_pdb_filename(self) -> list[pathlib.Path]:
         """
         Get a list of paths to the pdb files of the pre-minimized system.
 
         Returns
         -------
-        pdbs : list
-          list of paths (pathlib.PurePath) to the pdb files
+        pdbs : list[pathlib.Path]
+          list of paths (pathlib.Path) to the pdb files
         """
         pdbs = [pus[0].outputs['system_pdb'] for pus in self.data.values()]