Restrict to energy trajectory parquet files.

Author: lohedges

src/BioSimSpace/FreeEnergy/_relative.py

@@ -443,7 +443,7 @@ def analyse(work_dir, estimator="MBAR", method="alchemlyb", **kwargs):
 
         function_glob_dict = {
             "SOMD": (Relative._analyse_somd, "**/simfile.dat"),
-            "SOMD2": (Relative._analyse_somd2, "**/*.parquet"),
+            "SOMD2": (Relative._analyse_somd2, "**/energy_traj_*.parquet"),
             "GROMACS": (Relative._analyse_gromacs, "**/[!bar]*.xvg"),
             "AMBER": (Relative._analyse_amber, "**/*.out"),
         }
@@ -1958,7 +1958,7 @@ def _analyse_somd2(work_dir=None, estimator="MBAR", method="alchemlyb", **kwargs
 
         # Glob the data files.
         glob_path = _pathlib.Path(work_dir)
-        files = sorted(glob_path.glob("**/*.parquet"))
+        files = sorted(glob_path.glob("**/energy_traj_*.parquet"))
 
         # Loop over each file and try to extract the metadata to work out
         # the lambda value and temperature for each window.