Allow sxs.load to work on anything in simulations

sxs-collaboration · Oct 10, 2024 · 2f9b50f · 2f9b50f
1 parent 0fdf399
commit 2f9b50f
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/sxs/handlers.py b/sxs/handlers.py
@@ -298,6 +298,17 @@ def load(location, download=None, cache=None, progress=None, truepath=None, **kw
             return Simulation(location, download=download, cache=cache, progress=progress, **kwargs)
 
         else:
+            # Try to find an appropriate SXS file in the simulations
+            simulations = Simulations.load(
+                download=download,
+                local=kwargs.get("local", False),
+                annex_dir=kwargs.get("annex_dir", None)
+            )
+            if location.split("/Lev")[0] in simulations:
+                return Simulation(
+                    location, download=download, cache=cache, progress=progress, **kwargs
+                )
+
             # Try to find an appropriate SXS file in the catalog
             catalog = Catalog.load(download=download)
             selections = catalog.select_files(location)